Spaces:

neiro4droid
/

single-image-generator

Runtime error

neiro4droid commited on Sep 4, 2023

Commit

8a6d670

1 Parent(s): ed68d30

Update app.py

Files changed (1) hide show

app.py CHANGED Viewed

@@ -6,7 +6,7 @@ from diffusers import DiffusionPipeline
 def generate(prompt):
     pipeline = DiffusionPipeline.from_pretrained("Lykon/DreamShaper")
-    pipeline.to("cuda")
     return pipeline(prompt).images[0]
 iface = gr.Interface(fn=generate, inputs="text", outputs="image")

 def generate(prompt):
     pipeline = DiffusionPipeline.from_pretrained("Lykon/DreamShaper")
+    pipeline.to("cpu")
     return pipeline(prompt).images[0]
 iface = gr.Interface(fn=generate, inputs="text", outputs="image")