We read every piece of feedback, and take your input very seriously.
To see all available qualifiers, see our documentation.
There was an error while loading. Please reload this page.
DenseActivation
1 parent 144aec1 commit 0d1b698Copy full SHA for 0d1b698
backends/candle/src/models/dense.rs
@@ -4,10 +4,13 @@ use candle_nn::VarBuilder;
4
use serde::Deserialize;
5
6
#[derive(Debug, Clone, Deserialize, PartialEq)]
7
+/// The activation functions in `2_Dense/config.json` are defined as PyTorch imports
8
pub enum DenseActivation {
9
#[serde(rename = "torch.nn.modules.activation.Tanh")]
10
+ /// e.g. https://huggingface.co/sentence-transformers/LaBSE/blob/main/2_Dense/config.json
11
Tanh,
12
#[serde(rename = "torch.nn.modules.linear.Identity")]
13
+ /// e.g. https://huggingface.co/NovaSearch/stella_en_400M_v5/blob/main/2_Dense/config.json
14
Identity,
15
}
16
0 commit comments