license: mit | |
tags: | |
- pytorch | |
- diffusers | |
- unconditional-image-generation | |
- diffusion-models-class | |
# Example Fine-Tuned Model for Unit 2 of the [Diffusion Models Class 🧨](https://github.com/huggingface/diffusion-models-class) | |
The runwayml/stable-diffusion-v1-5 model fine-tuned on the FairFaces dataset to increase diversity. Part of DGM project | |
## Usage | |
```python | |
from diffusers import DDPMPipeline | |
pipeline = DDPMPipeline.from_pretrained('maa5iv/DGM-Fine-Tuned-Model-Project') | |
image = pipeline().images[0] | |
image | |
``` | |