Spaces:

Axolotlily
/

LDMtest3

Runtime error

Axolotlily commited on Aug 23, 2022

Commit

affed7f

1 Parent(s): 5a2144b

Update app.py

Files changed (1) hide show

app.py CHANGED Viewed

@@ -5,9 +5,9 @@ ldm = DiffusionPipeline.from_pretrained("CompVis/ldm-text2im-large-256")
 def inference(prompt):
     images = ldm([prompt],
-                      num_inference_steps=10,
                       eta=0.3,
                       guidance_scale=6)["sample"]
     return images
 gr.Interface(inference, "text", gr.Gallery()).launch()

 def inference(prompt):
     images = ldm([prompt],
+                      num_inference_steps=15,
                       eta=0.3,
                       guidance_scale=6)["sample"]
     return images
 gr.Interface(inference, "text", gr.Gallery()).launch()