map mistral model name to mixtral

Signed-off-by: Charlie Doern <[email protected]>
instructlab · Oct 18, 2024 · d237ea2 · d237ea2
1 parent 4bd07f5
commit d237ea2
Showing 1 changed file with 1 addition and 0 deletions.
diff --git a/src/instructlab/sdg/utils/models.py b/src/instructlab/sdg/utils/models.py
@@ -16,6 +16,7 @@
 # Map model names to their family
 MODEL_FAMILY_MAPPINGS = {
     "granite": "merlinite",
+    "mistral": "mixtral",
 }