diff options
| -rw-r--r-- | candle-transformers/src/models/mistral.rs | 3 |
1 files changed, 2 insertions, 1 deletions
diff --git a/candle-transformers/src/models/mistral.rs b/candle-transformers/src/models/mistral.rs index 0ba556bd..fe3187b9 100644 --- a/candle-transformers/src/models/mistral.rs +++ b/candle-transformers/src/models/mistral.rs @@ -40,7 +40,8 @@ impl Config { } // https://huggingface.co/Open-Orca/Mistral-7B-OpenOrca/blob/main/config.json - pub fn config_open_orca_chat_ml(use_flash_attn: bool) -> Self { + // https://huggingface.co/teknium/OpenHermes-2.5-Mistral-7B/blob/main/config.json + pub fn config_chat_ml(use_flash_attn: bool) -> Self { Self { vocab_size: 32002, hidden_size: 4096, |