/modular-diffusion

Python library for designing and training your own Diffusion Models with PyTorch.

Primary LanguagePythonMIT LicenseMIT

Watchers