Alex-23 commited on
Commit
1aad0b5
·
1 Parent(s): 680c81a

Update app.py

Browse files
Files changed (1) hide show
  1. app.py +0 -13
app.py CHANGED
@@ -1,13 +0,0 @@
1
- from diffusers import DDPMPipeline
2
- image_pipe = DDPMPipeline.from_pretrained("google/ddpm-celebahq-256")
3
- image_pipe.to("cuda")
4
- images = image_pipe().images
5
- image_pipe
6
- from diffusers import UNet2DModel
7
- repo_id = "google/ddpm-church-256"
8
- model = UNet2DModel.from_pretrained(repo_id)
9
- model
10
- model.config
11
- model_random = UNet2DModel(**model.config)
12
- model_random.save_pretrained("my_model")
13
- model_random = UNet2DModel.from_pretrained("my_model")