sd-aurora-128px / README.md
alkiskoudounas's picture
Upload README.md with huggingface_hub
7cc174f
|
raw
history blame
667 Bytes
metadata
license: mit
tags:
  - pytorch
  - diffusers
  - unconditional-image-generation
  - diffusion-models-class

Model Card for Stable Diffusion - Butterflies, 128px

Model developed for the Unit 1 of the Diffusion Models Class 🧨.

This model is a diffusion model for unconditional image generation of aurora borealis 🌌.
It is trained on a small collection aurora pictures and trained for 50 epochs, with 🤗 Accelerate.

Usage

from diffusers import DDPMPipeline

pipeline = DDPMPipeline.from_pretrained('alkiskoudounas/stable-diffusion-aurora-128px')
image = pipeline().images[0]
image