crowsonkb/k-diffusion

This is stars and forks stats for /crowsonkb/k-diffusion repository. As of 03 May, 2024 this repository has 1486 stars and 294 forks.

k-diffusion An implementation of Elucidating the Design Space of Diffusion-Based Generative Models (Karras et al., 2022) for PyTorch, with enhancements and additional features, such as improved sampling algorithms and transformer-based diffusion models. Installation k-diffusion can be installed via PyPI (pip install k-diffusion) but it will not include training and inference scripts, only library code that others can depend on. To run the training and inference scripts, clone this repository and...