liuliu/swift-diffusion

This is stars and forks stats for /liuliu/swift-diffusion repository. As of 30 Apr, 2024 this repository has 361 stars and 30 forks.

Swift Diffusion This is a single-file re-implementation of Stable Diffusion model. It includes CLIP text tokenizer, the models for CLIP text encoder, UNet diffusion model and the decoder model. It also includes PLMS inference implementation. The implementation tries to match the Stable Diffusion outputs layer-by-layer, thus, given the same start point x_T, this implementation and Stable Diffusion will output the same image. Rationale This re-implementation serves as an education for me to understand...