| license: mit | |
| tags: | |
| - pytorch | |
| - diffusers | |
| - unconditional-image-generation | |
| - diffusion-models-class | |
| # Model Card for Unit 1 of the [Diffusion Models Class 🧨](https://github.com/huggingface/diffusion-models-class) | |
| This model is a diffusion model for unconditional image generation of cute 🦋. | |
| The model was trained with Adam parameters from fast.ai. | |
| Batch size was also doubled to 64. | |
| Learning rate happens over 160 steps, aka 20% of training. | |
| ## Usage | |
| ```python | |
| from diffusers import DDPMPipeline | |
| pipeline = DDPMPipeline.from_pretrained('enzokro/sd-class-butterflies-64') | |
| image = pipeline().images[0] | |
| image | |
| ``` | |