├── .github └── workflows │ └── validate.yml ├── .gitignore ├── .zed └── tasks.json ├── Cargo.toml ├── LICENSE-MIT ├── LICENSE_APACHE ├── README.md └── src ├── lib.rs ├── models ├── attention.rs ├── embeddings.rs ├── mod.rs ├── resnet.rs └── unet_2d_blocks.rs ├── pipelines ├── mod.rs └── stable_diffusion.rs ├── transformers ├── clip.rs └── mod.rs └── utils.rs /.github/workflows/validate.yml: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/oxideai/diffusers-burn/HEAD/.github/workflows/validate.yml -------------------------------------------------------------------------------- /.gitignore: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/oxideai/diffusers-burn/HEAD/.gitignore -------------------------------------------------------------------------------- /.zed/tasks.json: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/oxideai/diffusers-burn/HEAD/.zed/tasks.json -------------------------------------------------------------------------------- /Cargo.toml: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/oxideai/diffusers-burn/HEAD/Cargo.toml -------------------------------------------------------------------------------- /LICENSE-MIT: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/oxideai/diffusers-burn/HEAD/LICENSE-MIT -------------------------------------------------------------------------------- /LICENSE_APACHE: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/oxideai/diffusers-burn/HEAD/LICENSE_APACHE -------------------------------------------------------------------------------- /README.md: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/oxideai/diffusers-burn/HEAD/README.md -------------------------------------------------------------------------------- /src/lib.rs: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/oxideai/diffusers-burn/HEAD/src/lib.rs -------------------------------------------------------------------------------- /src/models/attention.rs: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/oxideai/diffusers-burn/HEAD/src/models/attention.rs -------------------------------------------------------------------------------- /src/models/embeddings.rs: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/oxideai/diffusers-burn/HEAD/src/models/embeddings.rs -------------------------------------------------------------------------------- /src/models/mod.rs: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/oxideai/diffusers-burn/HEAD/src/models/mod.rs -------------------------------------------------------------------------------- /src/models/resnet.rs: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/oxideai/diffusers-burn/HEAD/src/models/resnet.rs -------------------------------------------------------------------------------- /src/models/unet_2d_blocks.rs: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/oxideai/diffusers-burn/HEAD/src/models/unet_2d_blocks.rs -------------------------------------------------------------------------------- /src/pipelines/mod.rs: -------------------------------------------------------------------------------- 1 | pub mod stable_diffusion; 2 | -------------------------------------------------------------------------------- /src/pipelines/stable_diffusion.rs: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/oxideai/diffusers-burn/HEAD/src/pipelines/stable_diffusion.rs -------------------------------------------------------------------------------- /src/transformers/clip.rs: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/oxideai/diffusers-burn/HEAD/src/transformers/clip.rs -------------------------------------------------------------------------------- /src/transformers/mod.rs: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/oxideai/diffusers-burn/HEAD/src/transformers/mod.rs -------------------------------------------------------------------------------- /src/utils.rs: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/oxideai/diffusers-burn/HEAD/src/utils.rs --------------------------------------------------------------------------------