huggingface/candle
1
0
Fork 0
mirror of https://github.com/huggingface/candle.git synced 2025-06-16 18:48:51 +00:00
Code Issues Packages Projects Releases Wiki Activity

make fn name generic (#1459)

Browse Source 
Co-authored-by: Ubuntu <danielclough@users.noreply.github.com>
This commit is contained in:
Daniel Clough
2023-12-20 17:16:31 -08:00
committed by GitHub
parent 8ede5f4210
commit 563a79afa1
1 changed files with 2 additions and 1 deletions
Download Patch File Download Diff File Expand all files Collapse all files

3
candle-transformers/src/models/mistral.rs
View File

@ -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,