Skip to content

Commit 0d1b698

Browse files
committed
Add comments in DenseActivation
1 parent 144aec1 commit 0d1b698

File tree

1 file changed

+3
-0
lines changed

1 file changed

+3
-0
lines changed

backends/candle/src/models/dense.rs

Lines changed: 3 additions & 0 deletions
Original file line numberDiff line numberDiff line change
@@ -4,10 +4,13 @@ use candle_nn::VarBuilder;
44
use serde::Deserialize;
55

66
#[derive(Debug, Clone, Deserialize, PartialEq)]
7+
/// The activation functions in `2_Dense/config.json` are defined as PyTorch imports
78
pub enum DenseActivation {
89
#[serde(rename = "torch.nn.modules.activation.Tanh")]
10+
/// e.g. https://huggingface.co/sentence-transformers/LaBSE/blob/main/2_Dense/config.json
911
Tanh,
1012
#[serde(rename = "torch.nn.modules.linear.Identity")]
13+
/// e.g. https://huggingface.co/NovaSearch/stella_en_400M_v5/blob/main/2_Dense/config.json
1114
Identity,
1215
}
1316

0 commit comments

Comments
 (0)