diff options
Diffstat (limited to 'candle-transformers/src/models/wuerstchen/mod.rs')
| -rw-r--r-- | candle-transformers/src/models/wuerstchen/mod.rs | 9 |
1 files changed, 9 insertions, 0 deletions
diff --git a/candle-transformers/src/models/wuerstchen/mod.rs b/candle-transformers/src/models/wuerstchen/mod.rs index 7b076f06..9bb37a3b 100644 --- a/candle-transformers/src/models/wuerstchen/mod.rs +++ b/candle-transformers/src/models/wuerstchen/mod.rs @@ -1,3 +1,12 @@ +//! Würstchen Efficient Diffusion Model +//! +//! 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) +//! pub mod attention_processor; pub mod common; pub mod ddpm; |