Repair a default value of step arg.

2024-07-21 12:13:08 +09:00 · 2024-07-21 12:13:08 +09:00 · 648e22a707
commit 648e22a707
parent 9860282397
1 changed files with 1 additions and 1 deletions
--- a/app/stable_diffusion_xl.py
+++ b/app/stable_diffusion_xl.py
@ -111,7 +111,7 @@ class SDXLTxt2Img:
            image = self.refiner(
                prompt=prompt,
                negative_prompt=n_prompt,
-                num_inference_steps=50,
+                num_inference_steps=steps,
                strength=0.3,
                guidance_scale=7.5,
                generator=generator,