add config_amazon_mistral_lite (#1493)

Co-authored-by: Ubuntu <danielclough@users.noreply.github.com>
This commit is contained in:
Daniel Clough 2023-12-28 10:59:58 -08:00 committed by GitHub
parent d35f0a1376
commit cd889c0f8a
No known key found for this signature in database
GPG Key ID: 4AEE18F83AFDEB23
1 changed files with 18 additions and 0 deletions

View File

@ -57,6 +57,24 @@ impl Config {
use_flash_attn,
}
}
// https://huggingface.co/amazon/MistralLite/blob/main/config.json
pub fn config_amazon_mistral_lite(use_flash_attn: bool) -> Self {
Self {
vocab_size: 32003,
hidden_size: 4096,
intermediate_size: 14336,
num_hidden_layers: 32,
num_attention_heads: 32,
num_key_value_heads: 8,
hidden_act: Activation::Silu,
max_position_embeddings: 32768,
rms_norm_eps: 1e-5,
rope_theta: 10_000.,
sliding_window: 4096,
use_flash_attn,
}
}
}
#[derive(Debug, Clone)]