https://github.com/drawcodeboy/ddpm
Denoising Diffusion Probabilistic Model (DDPM) Implementation (PyTorch)
https://github.com/drawcodeboy/ddpm
ddpm diffusion generative-ai python pytorch
Last synced: 3 months ago
JSON representation
Denoising Diffusion Probabilistic Model (DDPM) Implementation (PyTorch)
- Host: GitHub
- URL: https://github.com/drawcodeboy/ddpm
- Owner: drawcodeboy
- Created: 2025-01-11T13:54:58.000Z (about 1 year ago)
- Default Branch: main
- Last Pushed: 2025-02-20T02:39:23.000Z (about 1 year ago)
- Last Synced: 2025-02-20T03:28:32.760Z (about 1 year ago)
- Topics: ddpm, diffusion, generative-ai, python, pytorch
- Language: Python
- Homepage:
- Size: 168 MB
- Stars: 1
- Watchers: 1
- Forks: 0
- Open Issues: 0
-
Metadata Files:
- Readme: README.md
Awesome Lists containing this project
README
# DDPM Implementation with PyTorch
## Preview
## Description
* DDPM Implementation with PyTorch.
* The tests directory is created to check if the functions of DDPM are working properly and for analysis purposes.
## Setting
```
# Create Virtual Environment
python -m venv .venv
# Activate Virtual Environment
.venv\Scripts\activate # Windows
source .venv/bin/activate # Linux
# Install Packages
pip install -r requirements.txt
# Train
python train.py --config=base
# Sampling
python infer.py --config=base
```
## References
1. DPM: Sohl-Dickstein, Jascha, et al. "Deep unsupervised learning using nonequilibrium thermodynamics." International conference on machine learning. PMLR, 2015.
2. DDPM: Ho, Jonathan, Ajay Jain, and Pieter Abbeel. "Denoising diffusion probabilistic models." Advances in neural information processing systems 33 (2020): 6840-6851.
3. Original Repository: @lucidrains, denoising-diffusion-pytorch