├── .github └── workflows │ └── rust-ci.yml ├── .gitignore ├── Cargo.toml ├── LICENSE-APACHE ├── LICENSE-MIT ├── README.md ├── clippy.toml ├── examples ├── controlnet │ └── main.rs ├── stable-diffusion-img2img │ └── main.rs ├── stable-diffusion-inpaint │ └── main.rs ├── stable-diffusion │ └── main.rs └── tensor-tools.rs ├── media ├── in_img2img.jpg ├── out_img2img.jpg ├── out_inpaint.jpg ├── robot11.jpg ├── robot13.jpg ├── robot3.jpg ├── robot4.jpg ├── robot7.jpg ├── robot8.jpg ├── vermeer-edges.png ├── vermeer-out1.jpg ├── vermeer-out2.jpg ├── vermeer-out3.jpg ├── vermeer-out4.jpg ├── vermeer-out5.jpg └── vermeer.jpg ├── rustfmt.toml ├── scripts ├── download_weights_1.5.sh ├── download_weights_2.1.sh └── get_weights.py └── src ├── lib.rs ├── models ├── attention.rs ├── controlnet.rs ├── embeddings.rs ├── mod.rs ├── resnet.rs ├── unet_2d.rs ├── unet_2d_blocks.rs └── vae.rs ├── pipelines ├── mod.rs └── stable_diffusion.rs ├── schedulers ├── ddim.rs ├── ddpm.rs ├── dpmsolver_multistep.rs ├── euler_ancestral_discrete.rs ├── euler_discrete.rs ├── heun_discrete.rs ├── integrate.rs ├── k_dpm_2_ancestral_discrete.rs ├── k_dpm_2_discrete.rs ├── lms_discrete.rs ├── mod.rs └── pndm.rs ├── transformers ├── clip.rs └── mod.rs └── utils.rs /.github/workflows/rust-ci.yml: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/LaurentMazare/diffusers-rs/HEAD/.github/workflows/rust-ci.yml -------------------------------------------------------------------------------- /.gitignore: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/LaurentMazare/diffusers-rs/HEAD/.gitignore -------------------------------------------------------------------------------- /Cargo.toml: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/LaurentMazare/diffusers-rs/HEAD/Cargo.toml -------------------------------------------------------------------------------- /LICENSE-APACHE: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/LaurentMazare/diffusers-rs/HEAD/LICENSE-APACHE -------------------------------------------------------------------------------- /LICENSE-MIT: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/LaurentMazare/diffusers-rs/HEAD/LICENSE-MIT -------------------------------------------------------------------------------- /README.md: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/LaurentMazare/diffusers-rs/HEAD/README.md -------------------------------------------------------------------------------- /clippy.toml: -------------------------------------------------------------------------------- 1 | too-many-arguments-threshold = 20 2 | -------------------------------------------------------------------------------- /examples/controlnet/main.rs: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/LaurentMazare/diffusers-rs/HEAD/examples/controlnet/main.rs -------------------------------------------------------------------------------- /examples/stable-diffusion-img2img/main.rs: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/LaurentMazare/diffusers-rs/HEAD/examples/stable-diffusion-img2img/main.rs -------------------------------------------------------------------------------- /examples/stable-diffusion-inpaint/main.rs: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/LaurentMazare/diffusers-rs/HEAD/examples/stable-diffusion-inpaint/main.rs -------------------------------------------------------------------------------- /examples/stable-diffusion/main.rs: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/LaurentMazare/diffusers-rs/HEAD/examples/stable-diffusion/main.rs -------------------------------------------------------------------------------- /examples/tensor-tools.rs: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/LaurentMazare/diffusers-rs/HEAD/examples/tensor-tools.rs -------------------------------------------------------------------------------- /media/in_img2img.jpg: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/LaurentMazare/diffusers-rs/HEAD/media/in_img2img.jpg -------------------------------------------------------------------------------- /media/out_img2img.jpg: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/LaurentMazare/diffusers-rs/HEAD/media/out_img2img.jpg -------------------------------------------------------------------------------- /media/out_inpaint.jpg: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/LaurentMazare/diffusers-rs/HEAD/media/out_inpaint.jpg -------------------------------------------------------------------------------- /media/robot11.jpg: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/LaurentMazare/diffusers-rs/HEAD/media/robot11.jpg -------------------------------------------------------------------------------- /media/robot13.jpg: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/LaurentMazare/diffusers-rs/HEAD/media/robot13.jpg -------------------------------------------------------------------------------- /media/robot3.jpg: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/LaurentMazare/diffusers-rs/HEAD/media/robot3.jpg -------------------------------------------------------------------------------- /media/robot4.jpg: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/LaurentMazare/diffusers-rs/HEAD/media/robot4.jpg -------------------------------------------------------------------------------- /media/robot7.jpg: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/LaurentMazare/diffusers-rs/HEAD/media/robot7.jpg -------------------------------------------------------------------------------- /media/robot8.jpg: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/LaurentMazare/diffusers-rs/HEAD/media/robot8.jpg -------------------------------------------------------------------------------- /media/vermeer-edges.png: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/LaurentMazare/diffusers-rs/HEAD/media/vermeer-edges.png -------------------------------------------------------------------------------- /media/vermeer-out1.jpg: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/LaurentMazare/diffusers-rs/HEAD/media/vermeer-out1.jpg -------------------------------------------------------------------------------- /media/vermeer-out2.jpg: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/LaurentMazare/diffusers-rs/HEAD/media/vermeer-out2.jpg -------------------------------------------------------------------------------- /media/vermeer-out3.jpg: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/LaurentMazare/diffusers-rs/HEAD/media/vermeer-out3.jpg -------------------------------------------------------------------------------- /media/vermeer-out4.jpg: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/LaurentMazare/diffusers-rs/HEAD/media/vermeer-out4.jpg -------------------------------------------------------------------------------- /media/vermeer-out5.jpg: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/LaurentMazare/diffusers-rs/HEAD/media/vermeer-out5.jpg -------------------------------------------------------------------------------- /media/vermeer.jpg: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/LaurentMazare/diffusers-rs/HEAD/media/vermeer.jpg -------------------------------------------------------------------------------- /rustfmt.toml: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/LaurentMazare/diffusers-rs/HEAD/rustfmt.toml -------------------------------------------------------------------------------- /scripts/download_weights_1.5.sh: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/LaurentMazare/diffusers-rs/HEAD/scripts/download_weights_1.5.sh -------------------------------------------------------------------------------- /scripts/download_weights_2.1.sh: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/LaurentMazare/diffusers-rs/HEAD/scripts/download_weights_2.1.sh -------------------------------------------------------------------------------- /scripts/get_weights.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/LaurentMazare/diffusers-rs/HEAD/scripts/get_weights.py -------------------------------------------------------------------------------- /src/lib.rs: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/LaurentMazare/diffusers-rs/HEAD/src/lib.rs -------------------------------------------------------------------------------- /src/models/attention.rs: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/LaurentMazare/diffusers-rs/HEAD/src/models/attention.rs -------------------------------------------------------------------------------- /src/models/controlnet.rs: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/LaurentMazare/diffusers-rs/HEAD/src/models/controlnet.rs -------------------------------------------------------------------------------- /src/models/embeddings.rs: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/LaurentMazare/diffusers-rs/HEAD/src/models/embeddings.rs -------------------------------------------------------------------------------- /src/models/mod.rs: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/LaurentMazare/diffusers-rs/HEAD/src/models/mod.rs -------------------------------------------------------------------------------- /src/models/resnet.rs: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/LaurentMazare/diffusers-rs/HEAD/src/models/resnet.rs -------------------------------------------------------------------------------- /src/models/unet_2d.rs: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/LaurentMazare/diffusers-rs/HEAD/src/models/unet_2d.rs -------------------------------------------------------------------------------- /src/models/unet_2d_blocks.rs: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/LaurentMazare/diffusers-rs/HEAD/src/models/unet_2d_blocks.rs -------------------------------------------------------------------------------- /src/models/vae.rs: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/LaurentMazare/diffusers-rs/HEAD/src/models/vae.rs -------------------------------------------------------------------------------- /src/pipelines/mod.rs: -------------------------------------------------------------------------------- 1 | //! # Pipelines 2 | 3 | pub mod stable_diffusion; 4 | -------------------------------------------------------------------------------- /src/pipelines/stable_diffusion.rs: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/LaurentMazare/diffusers-rs/HEAD/src/pipelines/stable_diffusion.rs -------------------------------------------------------------------------------- /src/schedulers/ddim.rs: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/LaurentMazare/diffusers-rs/HEAD/src/schedulers/ddim.rs -------------------------------------------------------------------------------- /src/schedulers/ddpm.rs: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/LaurentMazare/diffusers-rs/HEAD/src/schedulers/ddpm.rs -------------------------------------------------------------------------------- /src/schedulers/dpmsolver_multistep.rs: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/LaurentMazare/diffusers-rs/HEAD/src/schedulers/dpmsolver_multistep.rs -------------------------------------------------------------------------------- /src/schedulers/euler_ancestral_discrete.rs: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/LaurentMazare/diffusers-rs/HEAD/src/schedulers/euler_ancestral_discrete.rs -------------------------------------------------------------------------------- /src/schedulers/euler_discrete.rs: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/LaurentMazare/diffusers-rs/HEAD/src/schedulers/euler_discrete.rs -------------------------------------------------------------------------------- /src/schedulers/heun_discrete.rs: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/LaurentMazare/diffusers-rs/HEAD/src/schedulers/heun_discrete.rs -------------------------------------------------------------------------------- /src/schedulers/integrate.rs: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/LaurentMazare/diffusers-rs/HEAD/src/schedulers/integrate.rs -------------------------------------------------------------------------------- /src/schedulers/k_dpm_2_ancestral_discrete.rs: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/LaurentMazare/diffusers-rs/HEAD/src/schedulers/k_dpm_2_ancestral_discrete.rs -------------------------------------------------------------------------------- /src/schedulers/k_dpm_2_discrete.rs: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/LaurentMazare/diffusers-rs/HEAD/src/schedulers/k_dpm_2_discrete.rs -------------------------------------------------------------------------------- /src/schedulers/lms_discrete.rs: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/LaurentMazare/diffusers-rs/HEAD/src/schedulers/lms_discrete.rs -------------------------------------------------------------------------------- /src/schedulers/mod.rs: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/LaurentMazare/diffusers-rs/HEAD/src/schedulers/mod.rs -------------------------------------------------------------------------------- /src/schedulers/pndm.rs: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/LaurentMazare/diffusers-rs/HEAD/src/schedulers/pndm.rs -------------------------------------------------------------------------------- /src/transformers/clip.rs: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/LaurentMazare/diffusers-rs/HEAD/src/transformers/clip.rs -------------------------------------------------------------------------------- /src/transformers/mod.rs: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/LaurentMazare/diffusers-rs/HEAD/src/transformers/mod.rs -------------------------------------------------------------------------------- /src/utils.rs: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/LaurentMazare/diffusers-rs/HEAD/src/utils.rs --------------------------------------------------------------------------------