--pag* and --sdxl-refiner-pag* argument reconstruction fix for --outp…

…ut-configs/--output-metadata
Teriks · Nov 6, 2024 · 32c08b7 · 32c08b7
1 parent 3e48de7
commit 32c08b7
Showing 1 changed file with 4 additions and 4 deletions.
diff --git a/dgenerate/pipelinewrapper/wrapper.py b/dgenerate/pipelinewrapper/wrapper.py
@@ -1087,18 +1087,18 @@ def reconstruct_dgenerate_opts(self,
             opts.append(('--pag',))
         else:
             if args.pag_scale is not None:
-                opts.append(('--pag-scale', args.pag_scale))
+                opts.append(('--pag-scales', args.pag_scale))
             if args.pag_adaptive_scale is not None:
-                opts.append(('--pag-adaptive-scale', args.pag_adaptive_scale))
+                opts.append(('--pag-adaptive-scales', args.pag_adaptive_scale))
 
         if args.sdxl_refiner_pag_scale == 3.0 and \
                 args.sdxl_refiner_pag_adaptive_scale == 0.0:
             opts.append(('--sdxl-refiner-pag',))
         else:
             if args.sdxl_refiner_pag_scale is not None:
-                opts.append(('--sdxl-refiner-pag-scale', args.sdxl_refiner_pag_scale))
+                opts.append(('--sdxl-refiner-pag-scales', args.sdxl_refiner_pag_scale))
             if args.sdxl_refiner_pag_adaptive_scale is not None:
-                opts.append(('--sdxl-refiner-pag-adaptive-scale', args.sdxl_refiner_pag_adaptive_scale))
+                opts.append(('--sdxl-refiner-pag-adaptive-scales', args.sdxl_refiner_pag_adaptive_scale))
 
         if self._sdxl_refiner_scheduler is not None:
             if self._sdxl_refiner_scheduler != self._scheduler: