mirror of
https://github.com/huggingface/candle.git
synced 2025-06-16 10:38:54 +00:00
add config for mamba 2.8b model parameter (#1946)
* first commit * Make the mamba config public. --------- Co-authored-by: laurent <laurent.mazare@gmail.com>
This commit is contained in:
@ -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 {
|
||||
|
Reference in New Issue
Block a user