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.

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