summaryrefslogtreecommitdiff
path: root/candle-transformers/src/models/fastvit.rs
diff options
context:
space:
mode:
authorLaurent Mazare <laurent.mazare@gmail.com>2024-09-28 23:48:00 +0200
committerGitHub <noreply@github.com>2024-09-28 23:48:00 +0200
commit261ed65f36c3f66ab33335850797ff473cbe4dd0 (patch)
tree678383e6025639996ebe8444b207a285c1a5445d /candle-transformers/src/models/fastvit.rs
parent62525e83526465b2d7969c8d3d0213b491b7ccc8 (diff)
downloadcandle-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.rs12
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 {