nakajimayoshi commited on
Commit
7a2937b
·
1 Parent(s): 891ef27

Update README.md

Browse files
Files changed (1) hide show
  1. README.md +3 -3
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 DDIMPipeline
17
 
18
  model_id = "nakajimayoshi/ddpm-iris-256"
19
 
20
  # load model and scheduler
21
- ddim = DDIMPipeline.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 = ddim().images[0]
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")