Add mathstral in the examples. (#2339)
This commit is contained in:
parent
f25173d68b
commit
42891cc613
|
@ -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(),
|
||||
}
|
||||
}
|
||||
}
|
||||
|
|
Loading…
Reference in New Issue