summaryrefslogtreecommitdiff
path: root/candle-transformers/src/models/siglip.rs
diff options
context:
space:
mode:
Diffstat (limited to 'candle-transformers/src/models/siglip.rs')
-rw-r--r--candle-transformers/src/models/siglip.rs8
1 files changed, 8 insertions, 0 deletions
diff --git a/candle-transformers/src/models/siglip.rs b/candle-transformers/src/models/siglip.rs
index 63b6635d..20464014 100644
--- a/candle-transformers/src/models/siglip.rs
+++ b/candle-transformers/src/models/siglip.rs
@@ -1,3 +1,11 @@
+//! Siglip model implementation.
+//!
+//! Siglip architecture combining vision and language for zero-shot tasks.
+//!
+//! References:
+//! - [Model Card](https://huggingface.co/google/siglip-base-patch16-224)
+//!
+
use crate::models::clip::div_l2_norm;
use candle::{IndexOp, Module, Result, Tensor, D};
use candle_nn::{layer_norm, linear, LayerNorm, Linear, VarBuilder};