Model Card for Smithsonian128UNet

A pre-trained UNet on the Smithsonain Butterflies 128x128 dataset (HuggingFace).

Model Description

An EMA model saved at the conclusion of training (1000 epochs) of an unconditional diffusion model on the Smithsonian Butterflies 128x128 dataset. Trained using the Simple Diffusion paradigm

• Developed by: [Gian Favero]
• License: [Apache 2.0]

Model Sources [optional]

• Repository: [https://github.com/faverogian/simpleDiffusion]
• Paper [optional]: [https://arxiv.org/pdf/2301.11093]

How to Get Started with the Model

Full usage details can be found at the GitHub repository page.

Environmental Impact

Carbon emissions can be estimated using the Machine Learning Impact calculator presented in Lacoste et al. (2019).

• Hardware Type: 4x 16GB V100
• Hours used: 1 hour
• Compute Region: Canada