summaryrefslogtreecommitdiff
path: root/candle-transformers/src/models/hiera.rs
diff options
context:
space:
mode:
authorzachcp <zachcp@users.noreply.github.com>2024-11-18 08:19:23 -0500
committerGitHub <noreply@github.com>2024-11-18 14:19:23 +0100
commit386fd8abb4be23c125e8100fed932f17d356a160 (patch)
treed4964322db768d31e2e4c1949848315fcfd7cfa2 /candle-transformers/src/models/hiera.rs
parent12d7e7b1450f0c3f87c3cce3a2a1dd1674cb8fd7 (diff)
downloadcandle-386fd8abb4be23c125e8100fed932f17d356a160.tar.gz
candle-386fd8abb4be23c125e8100fed932f17d356a160.tar.bz2
candle-386fd8abb4be23c125e8100fed932f17d356a160.zip
Module Docs (#2624)
* update whisper * update llama2c * update t5 * update phi and t5 * add a blip model * qlamma doc * add two new docs * add docs and emoji * additional models * openclip * pixtral * edits on the model docs * update yu * update a fe wmore models * add persimmon * add model-level doc * names * update module doc * links in heira * remove empty URL * update more hyperlinks * updated hyperlinks * more links * Update mod.rs --------- Co-authored-by: Laurent Mazare <laurent.mazare@gmail.com>
Diffstat (limited to 'candle-transformers/src/models/hiera.rs')
-rw-r--r--candle-transformers/src/models/hiera.rs7
1 files changed, 3 insertions, 4 deletions
diff --git a/candle-transformers/src/models/hiera.rs b/candle-transformers/src/models/hiera.rs
index 39f8d639..98ad8257 100644
--- a/candle-transformers/src/models/hiera.rs
+++ b/candle-transformers/src/models/hiera.rs
@@ -1,9 +1,8 @@
-//! [Hiera] inference implementation based on timm.
+//! Hiera inference implementation based on timm.
//!
-//! See "[Hiera: A Hierarchical Vision Transformer without the Bells-and-Whistles]"
-//! [Hiera: A Hierarchical Vision Transformer without the Bells-and-Whistles]: https://arxiv.org/abs/2306.00989
//!
-//! [Hiera]: https://github.com/huggingface/pytorch-image-models/blob/main/timm/models/hiera.py
+//! - 💻 [Hiera](https://github.com/huggingface/pytorch-image-models/blob/main/timm/models/hiera.py)
+//! - 📝 [Paper](https://arxiv.org/abs/2306.00989). Hiera: A Hierarchical Vision Transformer without the Bells-and-Whistles
use candle::{Result, D};
use candle_nn::{conv2d, layer_norm, linear, ops::softmax, Conv2dConfig, Func, VarBuilder};