crowsonkb / k-diffusion

Karras et al. (2022) diffusion models for PyTorch
MIT License
2.18k stars 369 forks source link