Reduced steps

app.py

@@ -171,22 +171,22 @@ def inference(text, style, inference_step, guidance_scale, seed):
 
 title = "Stable Diffusion with Textual Inversion"
 description = "A simple Gradio interface to infer Stable Diffusion and generate images with different art style"
-examples = [["A sweet potato farm", 'Concept', 30, 0.5, 1],
-            ["Sky full of cotton candy", 'Realistic', 30, 1.5, 2],
-            ["Coffin full of jello", 'Line', 30, 2.5, 3],
-            ["Water skiing on a lake", 'Ricky', 30, 3.5, 4],
-            ["Super slippery noodles", 'Plane Scape', 30, 4.5, 5],
-            ["Beautiful sunset", 'Concept', 30, 5.5, 6],
-            ["A glittering gem", 'Realistic', 30, 6.5, 7],
-            ["River rafting", 'Line', 30, 7.5, 8],
-            ["A green tea", 'Ricky', 30, 8.5, 9],
-            ["Three sphered rocks", 'Plane Scape', 30, 9.5, 10]]
+examples = [["A sweet potato farm", 'Concept', 20, 0.5, 1],
+            ["Sky full of cotton candy", 'Realistic', 20, 1.5, 2],
+            ["Coffin full of jello", 'Line', 20, 2.5, 3],
+            ["Water skiing on a lake", 'Ricky', 20, 3.5, 4],
+            ["Super slippery noodles", 'Plane Scape', 20, 4.5, 5],
+            ["Beautiful sunset", 'Concept', 20, 5.5, 6],
+            ["A glittering gem", 'Realistic', 20, 6.5, 7],
+            ["River rafting", 'Line', 20, 7.5, 8],
+            ["A green tea", 'Ricky', 20, 8.5, 9],
+            ["Three sphered rocks", 'Plane Scape', 20, 9.5, 10]]
 
 demo = gr.Interface(inference, 
                     inputs = [gr.Textbox(label="Prompt", type="text"),
                               gr.Dropdown(label="Style", choices=['Concept', 'Realistic', 'Line', 
                                                                   'Ricky', 'Plane Scape'], value="Concept"), 
-                              gr.Slider(10, 50, 30, step = 10, label="Inference steps"),
+                              gr.Slider(10, 50, 20, step = 10, label="Inference steps"),
                               gr.Slider(1, 10, 7.5, step = 0.1, label="Guidance scale"),
                               gr.Slider(0, 10000, 1, step = 1, label="Seed")],
                     outputs= [gr.Image(width=320, height=320, label="Output SAM")],

requirements.txt

@@ -1,3 +1,4 @@
+accelerate==0.33.0
 aiofiles==23.2.1
 altair==5.3.0
 annotated-types==0.7.0