Add mathstral in the examples. (#2339)

huggingface · Jul 18, 2024 · 42891cc · 42891cc
1 parent f25173d
commit 42891cc
Showing 1 changed file with 3 additions and 0 deletions.
diff --git a/candle-examples/examples/mistral/main.rs b/candle-examples/examples/mistral/main.rs
@@ -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(),
                 }
             }
         }