diff options
| author | Daniel Clough <9276072+danielclough@users.noreply.github.com> | 2023-12-20 17:16:31 -0800 |
|---|---|---|
| committer | GitHub <noreply@github.com> | 2023-12-21 02:16:31 +0100 |
| commit | 563a79afa1c6a5ad75ab88aef805f79fb1559f73 (patch) | |
| tree | 58593278eb3309be44d62a017a0afa6237acb40b | |
| parent | 8ede5f42102326a6c0e1cf7b824d781018ad6350 (diff) | |
| download | candle-563a79afa1c6a5ad75ab88aef805f79fb1559f73.tar.gz candle-563a79afa1c6a5ad75ab88aef805f79fb1559f73.tar.bz2 candle-563a79afa1c6a5ad75ab88aef805f79fb1559f73.zip | |
make fn name generic (#1459)
Co-authored-by: Ubuntu <danielclough@users.noreply.github.com>
| -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, |