1 2 3 4 5 6 7 8 9 10 11
//! # Models //! //! A collection of models to be used in a diffusion loop. pub mod attention; pub mod controlnet; pub mod embeddings; pub mod resnet; pub mod unet_2d; pub mod unet_2d_blocks; pub mod vae;