nakajimayoshi
commited on
Commit
·
7a2937b
1
Parent(s):
891ef27
Update README.md
Browse files
README.md
CHANGED
@@ -13,15 +13,15 @@ Diffusion model trained on a public dataset of images from [image data resource]
|
|
13 |
![realsamples](samples/real_samples.png)
|
14 |
|
15 |
```py
|
16 |
-
from diffusers import
|
17 |
|
18 |
model_id = "nakajimayoshi/ddpm-iris-256"
|
19 |
|
20 |
# load model and scheduler
|
21 |
-
|
22 |
|
23 |
# run pipeline in inference (sample random noise and denoise)
|
24 |
-
image =
|
25 |
|
26 |
# save image
|
27 |
image.save("ddpm_generated_image.png")
|
|
|
13 |
![realsamples](samples/real_samples.png)
|
14 |
|
15 |
```py
|
16 |
+
from diffusers import DDPMPipeline
|
17 |
|
18 |
model_id = "nakajimayoshi/ddpm-iris-256"
|
19 |
|
20 |
# load model and scheduler
|
21 |
+
ddpm = DDPMPipeline.from_pretrained(model_id) # you can replace DDPMPipeline with DDIMPipeline or PNDMPipeline for faster inference
|
22 |
|
23 |
# run pipeline in inference (sample random noise and denoise)
|
24 |
+
image = ddpm().images[0]
|
25 |
|
26 |
# save image
|
27 |
image.save("ddpm_generated_image.png")
|