lanzhiwang commited on
Commit
15ad5d6
·
1 Parent(s): 29b8b77
Files changed (1) hide show
  1. app.py +1 -1
app.py CHANGED
@@ -35,7 +35,7 @@ def erzeuge_komplex(prompt):
35
  # pipeline = DiffusionPipeline.from_pretrained("google/ddpm-cat-256")
36
  pipeline = DiffusionPipeline.from_pretrained("google/ddpm-celebahq-256")
37
 
38
- pipeline.to("cuda")
39
 
40
 
41
  with gr.Blocks() as demo:
 
35
  # pipeline = DiffusionPipeline.from_pretrained("google/ddpm-cat-256")
36
  pipeline = DiffusionPipeline.from_pretrained("google/ddpm-celebahq-256")
37
 
38
+ # pipeline.to("cuda")
39
 
40
 
41
  with gr.Blocks() as demo: