huggingface/candle
1
0
Fork 0
mirror of https://github.com/huggingface/candle.git synced 2025-06-15 10:26:33 +00:00
Code Issues Packages Projects Releases Wiki Activity

Add mathstral in the examples. (#2339)

Browse Source
This commit is contained in:
Laurent Mazare
2024-07-18 07:24:49 +01:00
committed by GitHub
parent f25173d68b
commit 42891cc613
1 changed files with 3 additions and 0 deletions
Download Patch File Download Diff File Expand all files Collapse all files

3
candle-examples/examples/mistral/main.rs
View File

@ -147,6 +147,8 @@ enum Which {
Mistral7bInstructV01,
#[value(name = "7b-instruct-v0.2")]
Mistral7bInstructV02,
#[value(name = "7b-maths-v0.1")]
Mathstral7bV01,
}
#[derive(Parser, Debug)]
@ -266,6 +268,7 @@ fn main() -> Result<()> {
Which::Mistral7bV02 => "mistralai/Mistral-7B-v0.2".to_string(),
Which::Mistral7bInstructV01 => "mistralai/Mistral-7B-Instruct-v0.1".to_string(),
Which::Mistral7bInstructV02 => "mistralai/Mistral-7B-Instruct-v0.2".to_string(),
Which::Mathstral7bV01 => "mistralai/mathstral-7B-v0.1".to_string(),
}
}
}