diff options
author | Laurent Mazare <laurent.mazare@gmail.com> | 2024-09-28 23:48:00 +0200 |
---|---|---|
committer | GitHub <noreply@github.com> | 2024-09-28 23:48:00 +0200 |
commit | 261ed65f36c3f66ab33335850797ff473cbe4dd0 (patch) | |
tree | 678383e6025639996ebe8444b207a285c1a5445d /candle-transformers/src/models/fastvit.rs | |
parent | 62525e83526465b2d7969c8d3d0213b491b7ccc8 (diff) | |
download | candle-261ed65f36c3f66ab33335850797ff473cbe4dd0.tar.gz candle-261ed65f36c3f66ab33335850797ff473cbe4dd0.tar.bz2 candle-261ed65f36c3f66ab33335850797ff473cbe4dd0.zip |
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.
Diffstat (limited to 'candle-transformers/src/models/fastvit.rs')
-rw-r--r-- | candle-transformers/src/models/fastvit.rs | 12 |
1 files changed, 6 insertions, 6 deletions
diff --git a/candle-transformers/src/models/fastvit.rs b/candle-transformers/src/models/fastvit.rs index b7bdaf88..8eae8bb2 100644 --- a/candle-transformers/src/models/fastvit.rs +++ b/candle-transformers/src/models/fastvit.rs @@ -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 { |