sd-aurora-256px / README.md
alkiskoudounas's picture
Update README.md
c2a5e79
metadata
license: mit
tags:
  - pytorch
  - diffusers
  - unconditional-image-generation
  - diffusion-models-class

Model Card for Stable Diffusion - Aurora Borealis, 256px

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 of aurora pictures and trained for 50 epochs, with 🤗 Accelerate.

Usage

from diffusers import DDPMPipeline

pipeline = DDPMPipeline.from_pretrained('alkiskoudounas/sd-aurora-256px')

Example

Here you can find an example of the output of the model, in a batch of 8 images:

Aurora Borealis Example