mirror of
https://github.com/huggingface/candle.git
synced 2025-06-20 12:06:35 +00:00
Add the SigLIP model. (#2515)
* Add the SigLIP model. * Add more to the forward pass of the vision model. * Complete the forward pass. * Add the siglip example. * Fix. * Another fix. * Get everything in place. * Add a readme.
This commit is contained in:
@ -11,13 +11,13 @@ use candle_nn::{
|
||||
BatchNorm, Conv2d, Conv2dConfig, Func, VarBuilder,
|
||||
};
|
||||
|
||||
#[derive(Clone, Debug)]
|
||||
#[derive(serde::Serialize, serde::Deserialize, Clone, Debug)]
|
||||
pub struct Config {
|
||||
exp_ratio: usize,
|
||||
in_channels: usize,
|
||||
blocks: [usize; 4],
|
||||
attn: bool,
|
||||
lkc_use_act: bool,
|
||||
pub exp_ratio: usize,
|
||||
pub in_channels: usize,
|
||||
pub blocks: [usize; 4],
|
||||
pub attn: bool,
|
||||
pub lkc_use_act: bool,
|
||||
}
|
||||
|
||||
impl Config {
|
||||
|
Reference in New Issue
Block a user