huggingface/candle
1
0
Fork 0
mirror of https://github.com/huggingface/candle.git synced 2025-06-19 11:56:45 +00:00
Code Issues Packages Projects Releases Wiki Activity

Remove the unused pragma and properly apply the bias. (#1147)

Browse Source
This commit is contained in:
Laurent Mazare
2023-10-22 06:47:40 +01:00
committed by GitHub
parent 3115fe42e4
commit 5b32c2a41e
3 changed files with 15 additions and 22 deletions
Download Patch File Download Diff File Expand all files Collapse all files

8
candle-transformers/src/models/with_tracing.rs
View File

@ -32,6 +32,14 @@ pub struct Linear {
span: tracing::Span,
}
impl Linear {
pub fn from_weights(weights: Tensor, bias: Option<Tensor>) -> Self {
let inner = candle_nn::Linear::new(weights, bias);
let span = tracing::span!(tracing::Level::TRACE, "linear");
Self { inner, span }
}
}
pub fn linear(d1: usize, d2: usize, vb: VarBuilder) -> Result<Linear> {
let inner = candle_nn::linear(d1, d2, vb)?;
let span = tracing::span!(tracing::Level::TRACE, "linear");