Merge pull request #315 from cdoern/release-v0.3

map mistral model name to mixtral
instructlab · Oct 18, 2024 · 481e3f6 · 481e3f6
2 parents 4bd07f5 + d237ea2
commit 481e3f6
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",
 }