huggingface/candle
1
0
Fork 0
mirror of https://github.com/huggingface/candle.git synced 2025-06-21 12:20:46 +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

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

@ -1,4 +1,3 @@
#![allow(unused)]
use super::blip_text;
use super::with_tracing::{conv2d, linear, Conv2d, Linear};
use candle::{Module, Result, Tensor, D};
@ -65,7 +64,6 @@ struct VisionEmbeddings {
class_embedding: Tensor,
patch_embedding: Conv2d,
position_embedding: Tensor,
num_positions: usize,
}
impl VisionEmbeddings {
@ -91,7 +89,6 @@ impl VisionEmbeddings {
class_embedding,
patch_embedding,
position_embedding,
num_positions,
})
}
}
@ -117,8 +114,6 @@ struct Attention {
qkv: Linear,
projection: Linear,
scale: f64,
embed_dim: usize,
head_dim: usize,
num_heads: usize,
}
@ -134,8 +129,6 @@ impl Attention {
qkv,
projection,
scale,
embed_dim,
head_dim,
num_heads,
})
}