summaryrefslogtreecommitdiff
path: root/candle-transformers/src/models/stable_diffusion/mod.rs
diff options
context:
space:
mode:
Diffstat (limited to 'candle-transformers/src/models/stable_diffusion/mod.rs')
-rw-r--r--candle-transformers/src/models/stable_diffusion/mod.rs9
1 files changed, 9 insertions, 0 deletions
diff --git a/candle-transformers/src/models/stable_diffusion/mod.rs b/candle-transformers/src/models/stable_diffusion/mod.rs
index 37f4cdbf..d3e2032b 100644
--- a/candle-transformers/src/models/stable_diffusion/mod.rs
+++ b/candle-transformers/src/models/stable_diffusion/mod.rs
@@ -1,3 +1,12 @@
+//! Stable Diffusion
+//!
+//! Stable Diffusion is a latent text-to-image diffusion model capable of
+//! generating photo-realistic images given any text input.
+//!
+//! - [Original Repository](https://github.com/CompVis/stable-diffusion)
+//! - [Hugging Face](https://huggingface.co/runwayml/stable-diffusion-v1-5)
+//!
+
pub mod attention;
pub mod clip;
pub mod ddim;