diff options
Diffstat (limited to 'candle-transformers/src/models/wuerstchen/mod.rs')
| -rw-r--r-- | candle-transformers/src/models/wuerstchen/mod.rs | 13 |
1 files changed, 10 insertions, 3 deletions
diff --git a/candle-transformers/src/models/wuerstchen/mod.rs b/candle-transformers/src/models/wuerstchen/mod.rs index 9bb37a3b..ae42c4a8 100644 --- a/candle-transformers/src/models/wuerstchen/mod.rs +++ b/candle-transformers/src/models/wuerstchen/mod.rs @@ -3,10 +3,17 @@ //! Würstchen is an efficient diffusion model architecture for generating images using //! a two-stage approach with a small decoder and prior network. //! -//! - [Paper Link](https://openreview.net/pdf?id=gU58AyJlYz) -//! - [GH Link](https://github.com/dome272/Wuerstchen) -//! - [Reference Implementation](https://github.com/huggingface/diffusers/blob/main/src/diffusers/pipelines/wuerstchen/pipeline_wuerstchen.py) +//! - 💻 [GH Link](https://github.com/dome272/Wuerstchen) +//! - 🤗 [HF Link](https://github.com/huggingface/diffusers/blob/main/src/diffusers/pipelines/wuerstchen/pipeline_wuerstchen.py) +//! - 📝 [Paper](https://openreview.net/pdf?id=gU58AyJlYz) //! +//! ## Example +//! +//! <div align=center> +//! <img src="https://github.com/huggingface/candle/raw/main/candle-examples/examples/wuerstchen/assets/cat.jpg" alt="" width=320> +//! <p>"Anthropomorphic cat dressed as a fire fighter"</p> +//! </div> + pub mod attention_processor; pub mod common; pub mod ddpm; |