johnowhitaker's picture
Update README.md
5662182
|
raw
history blame
386 Bytes
Diffusion model trained on https://huggingface.co/datasets/huggan/smithsonian_butterflies_subset
Example outputs:
![examples](https://huggingface.co/tglcourse/ddpm-butterflies-64px/resolve/main/example_outputs.png)
Usage:
```
pipeline = DDPMPipeline.from_pretrained('tglcourse/ddpm-ema-butterflies-64').to(device)
images = pipeline(batch_size=8).images
```
More info in course soon.