diff options
| -rw-r--r-- | candle-transformers/src/models/mamba.rs | 8 |
1 files changed, 4 insertions, 4 deletions
diff --git a/candle-transformers/src/models/mamba.rs b/candle-transformers/src/models/mamba.rs index 597dd2cd..836327ee 100644 --- a/candle-transformers/src/models/mamba.rs +++ b/candle-transformers/src/models/mamba.rs @@ -10,10 +10,10 @@ const D_STATE: usize = 16; #[derive(Debug, Clone, serde::Deserialize)] pub struct Config { - d_model: usize, - n_layer: usize, - vocab_size: usize, - pad_vocab_size_multiple: usize, + pub d_model: usize, + pub n_layer: usize, + pub vocab_size: usize, + pub pad_vocab_size_multiple: usize, } impl Config { |