Skip to content

Designed DDPM, DDIM, cVAE Image Generation Models. Lightweight UNet Architecture for maintaining high detail while saving VRAM

Notifications You must be signed in to change notification settings

llLucidll/ProbabalisticDiffusion

Folders and files

NameName
Last commit message
Last commit date

Latest commit

 

History

9 Commits
 
 
 
 
 
 

Repository files navigation

Implemented as seen in Song et al.

  1. A lightweight Unet model with time embeddings for use with both Probabalistic and Deterministic Diffusion Models. Able to obtain 95+ accuracy on FashionMNIST generation

  2. Lightweight conditional VAE architecture with large latent space able to generalize well to FashionMNIST in just one epoch of training obtaining accuracies of 90+

About

Designed DDPM, DDIM, cVAE Image Generation Models. Lightweight UNet Architecture for maintaining high detail while saving VRAM

Resources

Stars

Watchers

Forks

Releases

No releases published

Packages

No packages published

Languages