pytorch diffusers unconditional-image-generation diffusion-models-class

Model Card for Stable Diffusion - Aurora Borealis, 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 of aurora pictures and trained for 50 epochs, with ๐Ÿค— Accelerate.

Usage

from diffusers import DDPMPipeline

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

Example

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

Butterflies Example