diff options
| author | Jorge António <matroid@outlook.com> | 2024-03-27 06:47:23 +0000 |
|---|---|---|
| committer | GitHub <noreply@github.com> | 2024-03-27 07:47:23 +0100 |
| commit | 75b6d4b0da4e7fef82d9f61e274b49af55777acf (patch) | |
| tree | 98c2c1e78a167f2ba89fea3c670f99a7e8290cbf | |
| parent | 66f0a4eeea02f069838903a18dd6402821e43271 (diff) | |
| download | candle-75b6d4b0da4e7fef82d9f61e274b49af55777acf.tar.gz candle-75b6d4b0da4e7fef82d9f61e274b49af55777acf.tar.bz2 candle-75b6d4b0da4e7fef82d9f61e274b49af55777acf.zip | |
add config for mamba 2.8b model parameter (#1946)
* first commit
* Make the mamba config public.
---------
Co-authored-by: laurent <laurent.mazare@gmail.com>
| -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 { |