FrankCCCCC commited on
Commit
6aa387f
·
verified ·
1 Parent(s): e273605

Correct the model name

Browse files
Files changed (1) hide show
  1. README.md +3 -3
README.md CHANGED
@@ -38,7 +38,7 @@ This model implements the DDPM approach described in the paper ["Denoising Diffu
38
  from diffusers import DDPMPipeline
39
 
40
  # Load the model
41
- model_id = "FrankCCCCC/ddpm-ema-cifar10" # Replace with actual repo ID
42
  pipeline = DDPMPipeline.from_pretrained(model_id)
43
 
44
  # Generate an image
@@ -51,7 +51,7 @@ image.save("generated_cifar10.png")
51
  ```python
52
  from diffusers import DDPMPipeline
53
 
54
- pipeline = DDPMPipeline.from_pretrained("FrankCCCCC/ddpm-ema-cifar10")
55
 
56
  # Generate batch of images
57
  images = pipeline(batch_size=4).images
@@ -66,7 +66,7 @@ for i, image in enumerate(images):
66
  ```python
67
  from diffusers import DDPMPipeline, DDIMScheduler, PNDMScheduler
68
 
69
- pipeline = DDPMPipeline.from_pretrained("FrankCCCCC/ddpm-ema-cifar10")
70
 
71
  # Use DDIM scheduler for faster inference
72
  ddim_scheduler = DDIMScheduler.from_config(pipeline.scheduler.config)
 
38
  from diffusers import DDPMPipeline
39
 
40
  # Load the model
41
+ model_id = "FrankCCCCC/ddpm_ema_cifar10" # Replace with actual repo ID
42
  pipeline = DDPMPipeline.from_pretrained(model_id)
43
 
44
  # Generate an image
 
51
  ```python
52
  from diffusers import DDPMPipeline
53
 
54
+ pipeline = DDPMPipeline.from_pretrained("FrankCCCCC/ddpm_ema_cifar10")
55
 
56
  # Generate batch of images
57
  images = pipeline(batch_size=4).images
 
66
  ```python
67
  from diffusers import DDPMPipeline, DDIMScheduler, PNDMScheduler
68
 
69
+ pipeline = DDPMPipeline.from_pretrained("FrankCCCCC/ddpm_ema_cifar10")
70
 
71
  # Use DDIM scheduler for faster inference
72
  ddim_scheduler = DDIMScheduler.from_config(pipeline.scheduler.config)