Om2005Prakash's picture
Create README.md
f4c89d3 verified
Github Repo https://github.com/kevalpithadiya/Diffusion-Models
# Diffusion Models
This work was part of a term paper / project for the UMC203 Introduction to AI and ML course at IISc.
## Contributors
- Om Prakash Choudhary
- Himesh Kumar Anant
- Jithendra Rao Kasibhatla
- Keval Pithadiya
## Models Trained
We trained Denoising Diffusion Probabilitic Models using the HuggingFace `diffusers` library for the clean image prediction and the noise prediction objectives on the below datasets and image sizes
- CIFAR10 (32x32)
- CelebA (32x32)
- CelebA (128x128)
- MNIST (32x32)
The trained models can be downloaded from this Google Drive link:
https://drive.google.com/drive/folders/1HmLK4hkkI6-Fl6vaHyX05PBe8U2FWl5E?usp=drive_link