summaryrefslogtreecommitdiff
path: root/candle-transformers/src/models/dinov2.rs
diff options
context:
space:
mode:
Diffstat (limited to 'candle-transformers/src/models/dinov2.rs')
-rw-r--r--candle-transformers/src/models/dinov2.rs5
1 files changed, 5 insertions, 0 deletions
diff --git a/candle-transformers/src/models/dinov2.rs b/candle-transformers/src/models/dinov2.rs
index 706dfda0..df8834d1 100644
--- a/candle-transformers/src/models/dinov2.rs
+++ b/candle-transformers/src/models/dinov2.rs
@@ -1,3 +1,8 @@
+//! Implementation of the DINOv2 models from Meta Research.
+//!
+//! See:
+//! - DINOv2: ["DINOv2: Learning Robust Visual Features without Supervision"](https://github.com/facebookresearch/dinov2)
+//!
use candle::{IndexOp, Result, Tensor, D};
use candle_nn::{layer_norm, LayerNorm, Linear, Module, VarBuilder};