fix strength error

kohya-ss · Dec 1, 2023 · 2952bca · 2952bca
1 parent 29b6fa6
commit 2952bca
Showing 1 changed file with 10 additions and 10 deletions.
diff --git a/sdxl_gen_img.py b/sdxl_gen_img.py
@@ -1829,20 +1829,20 @@ def __getattr__(self, item):
     # Gradual Latent
     if args.gradual_latent_timesteps is not None:
         if args.gradual_latent_unsharp_params:
-            ksize, sigma, strength = [float(v) for v in args.gradual_latent_unsharp_params.split(",")]
-            ksize = int(ksize)
+            us_ksize, us_sigma, us_strength = [float(v) for v in args.gradual_latent_unsharp_params.split(",")]
+            us_ksize = int(us_ksize)
         else:
-            ksize, sigma, strength = None, None, None
+            us_ksize, us_sigma, us_strength = None, None, None
 
         gradual_latent = GradualLatent(
             args.gradual_latent_ratio,
             args.gradual_latent_timesteps,
             args.gradual_latent_every_n_steps,
             args.gradual_latent_ratio_step,
             args.gradual_latent_s_noise,
-            ksize,
-            sigma,
-            strength,
+            us_ksize,
+            us_sigma,
+            us_strength,
         )
         pipe.set_gradual_latent(gradual_latent)
 
@@ -2650,12 +2650,12 @@ def scale_and_round(x):
                     if gl_timesteps < 0:
                         gl_timesteps = args.gradual_latent_timesteps or 650
                     if gl_unsharp_params is not None:
-                        ksize, sigma, strength = [float(v) for v in gl_unsharp_params.split(",")]
-                        ksize = int(ksize)
+                        us_ksize, us_sigma, us_strength = [float(v) for v in gl_unsharp_params.split(",")]
+                        us_ksize = int(us_ksize)
                     else:
-                        ksize, sigma, strength = None, None, None
+                        us_ksize, us_sigma, us_strength = None, None, None
                     gradual_latent = GradualLatent(
-                        gl_ratio, gl_timesteps, gl_every_n_steps, gl_ratio_step, gl_s_noise, ksize, sigma, strength
+                        gl_ratio, gl_timesteps, gl_every_n_steps, gl_ratio_step, gl_s_noise, us_ksize, us_sigma, us_strength
                     )
                     pipe.set_gradual_latent(gradual_latent)