diff options
| author | Radamés Ajna <radamajna@gmail.com> | 2023-10-25 23:24:02 -0700 |
|---|---|---|
| committer | GitHub <noreply@github.com> | 2023-10-26 07:24:02 +0100 |
| commit | e5dc8cb4f40f564bcc1acce4926aa99f25464681 (patch) | |
| tree | b66cc172e3333a72e8ca3d386567fa231935042c /candle-transformers/src/models/blip_text.rs | |
| parent | e7b886d56f05b30b7814e0aaf0aea505fcfe30d2 (diff) | |
| download | candle-e5dc8cb4f40f564bcc1acce4926aa99f25464681.tar.gz candle-e5dc8cb4f40f564bcc1acce4926aa99f25464681.tar.bz2 candle-e5dc8cb4f40f564bcc1acce4926aa99f25464681.zip | |
[Wasm] BLIP Example (#1183)
* blip wasm start
* fix dependency issue, move token stream here
* vanilla js worker
* roll back vscode
* spell
Diffstat (limited to 'candle-transformers/src/models/blip_text.rs')
| -rw-r--r-- | candle-transformers/src/models/blip_text.rs | 3 |
1 files changed, 2 insertions, 1 deletions
diff --git a/candle-transformers/src/models/blip_text.rs b/candle-transformers/src/models/blip_text.rs index 3a519e4c..1862abef 100644 --- a/candle-transformers/src/models/blip_text.rs +++ b/candle-transformers/src/models/blip_text.rs @@ -1,8 +1,9 @@ use super::with_tracing::{linear, Embedding, Linear}; use candle::{Module, Result, Tensor, D}; use candle_nn::{layer_norm, LayerNorm, VarBuilder}; +use serde::Deserialize; -#[derive(Debug, Clone)] +#[derive(Debug, Clone, Deserialize)] pub struct Config { pub vocab_size: usize, pub hidden_size: usize, |