Correct the model name
Browse files
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/
|
| 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/
|
| 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/
|
| 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)
|