lanzhiwang commited on
Commit
29b8b77
1 Parent(s): c4f95ed
Files changed (2) hide show
  1. Diffusers.ipynb +0 -0
  2. app.py +2 -1
Diffusers.ipynb ADDED
The diff for this file is too large to render. See raw diff
 
app.py CHANGED
@@ -32,7 +32,8 @@ def erzeuge_komplex(prompt):
32
 
33
 
34
  # pipeline = DiffusionPipeline.from_pretrained("stabilityai/stable-diffusion-2")
35
- pipeline = DiffusionPipeline.from_pretrained("google/ddpm-cat-256")
 
36
 
37
  pipeline.to("cuda")
38
 
 
32
 
33
 
34
  # pipeline = DiffusionPipeline.from_pretrained("stabilityai/stable-diffusion-2")
35
+ # pipeline = DiffusionPipeline.from_pretrained("google/ddpm-cat-256")
36
+ pipeline = DiffusionPipeline.from_pretrained("google/ddpm-celebahq-256")
37
 
38
  pipeline.to("cuda")
39