Spaces:

gradio
/

diffusers_with_batching

Runtime error

abidlabs HF Staff commited on Feb 17, 2023

Commit

ede6ba2

1 Parent(s): c02844f

Update app.py

Files changed (1) hide show

app.py CHANGED Viewed

@@ -2,7 +2,7 @@ from diffusers import DiffusionPipeline
 import gradio as gr
 generator = DiffusionPipeline.from_pretrained("CompVis/ldm-text2im-large-256")
-# generator.to("cuda")  # If you have a GPU, uncomment this line
 def generate(prompts):
   images = generator(list(prompts)).images

 import gradio as gr
 generator = DiffusionPipeline.from_pretrained("CompVis/ldm-text2im-large-256")
+generator.to("cuda")  # If you have a GPU, uncomment this line
 def generate(prompts):
   images = generator(list(prompts)).images