diff --git a/src/instructlab/sdg/utils/models.py b/src/instructlab/sdg/utils/models.py index b6b03b00..a8d531a7 100644 --- 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", }