| 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) | |
| In ddpm-celebahq-256, butterflies-256 1000sz datasets Finetune 10 epochs with learning rate 1e-5, 8 batch-size, | |
| ## Usage | |
| ```python | |
| from diffusers import DDPMPipeline | |
| pipeline = DDPMPipeline.from_pretrained('KoreadeepKai/ddpm-celebahq-finetuned-butterflies-10epochs') | |
| image = pipeline().images[0] | |
| image | |
| ``` | |