Added Inference Steps option

app.py

@@ -61,6 +61,7 @@ def generate(
     width: int = 1024,
     height: int = 1024,
     guidance_scale: float = 3,
+    num_inference_steps = 25,
     randomize_seed: bool = False,
     progress=gr.Progress(track_tqdm=True),
 ):
@@ -77,7 +78,7 @@ def generate(
         width=width,
         height=height,
         guidance_scale=guidance_scale,
-        num_inference_steps=25,
+        num_inference_steps=num_inference_steps,
         num_images_per_prompt=1,
         cross_attention_kwargs={"scale": 0.65},
         output_type="pil",
@@ -163,6 +164,13 @@ with gr.Blocks(css=css, theme="pseudolab/huggingface-korea-theme") as demo:
                 step=0.1,
                 value=6,
             )
+            num_inference_steps = gr.Slider(
+                label="Steps",
+                minimum=20,
+                maximum=40,
+                step=1,
+                value=25,
+            )
 
     gr.Examples(
         examples=examples,
@@ -195,6 +203,7 @@ with gr.Blocks(css=css, theme="pseudolab/huggingface-korea-theme") as demo:
             width,
             height,
             guidance_scale,
+            num_inference_steps,
             randomize_seed,
         ],
         outputs=[result, seed],