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

Make some model cloneable. (#1125)

Browse Source
This commit is contained in:
Laurent Mazare
2023-10-18 19:30:47 +01:00
committed by GitHub
parent 620c94d12e
commit 185b54a33b
5 changed files with 25 additions and 20 deletions
Download Patch File Download Diff File Expand all files Collapse all files

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

@ -1,7 +1,7 @@
use candle::{Module, Result, Tensor};
use candle_nn::VarBuilder;
#[derive(Debug)]
#[derive(Debug, Clone)]
pub struct Embedding {
inner: candle_nn::Embedding,
span: tracing::Span,
@ -26,7 +26,7 @@ impl Module for Embedding {
}
}
#[derive(Debug)]
#[derive(Debug, Clone)]
pub struct Linear {
inner: candle_nn::Linear,
span: tracing::Span,
@ -52,7 +52,7 @@ impl Module for Linear {
}
// Wrap the conv2d op to provide some tracing.
#[derive(Debug)]
#[derive(Debug, Clone)]
pub struct Conv2d {
inner: candle_nn::Conv2d,
span: tracing::Span,
@ -78,6 +78,7 @@ pub fn conv2d(
}
// QMatMul wrapper adding some tracing.
#[derive(Clone)]
pub struct QMatMul {
inner: candle::quantized::QMatMul,
span: tracing::Span,