dev: mistral-small

zhudotexe · Oct 2, 2024 · 8c8c1b5 · 8c8c1b5
1 parent 4334758
commit 8c8c1b5
Showing 1 changed file with 2 additions and 1 deletion.
diff --git a/sandbox/mistral03-fc.py b/sandbox/mistral03-fc.py
@@ -5,7 +5,8 @@
 from kani.engines.huggingface import HuggingEngine
 from kani.prompts.impl.mistral import MISTRAL_V3_PIPELINE, MistralFunctionCallingAdapter
 
-model = HuggingEngine(model_id="mistralai/Mistral-7B-Instruct-v0.3", prompt_pipeline=MISTRAL_V3_PIPELINE)
+# model = HuggingEngine(model_id="mistralai/Mistral-7B-Instruct-v0.3", prompt_pipeline=MISTRAL_V3_PIPELINE)
+model = HuggingEngine(model_id="mistralai/Mistral-Small-Instruct-2409", prompt_pipeline=MISTRAL_V3_PIPELINE)
 engine = MistralFunctionCallingAdapter(model)