From d237ea23ce7cdfbce68201c143ca460ea214c0a6 Mon Sep 17 00:00:00 2001
From: Charlie Doern <cdoern@redhat.com>
Date: Fri, 18 Oct 2024 16:47:45 -0400
Subject: [PATCH] map mistral model name to mixtral

Signed-off-by: Charlie Doern <cdoern@redhat.com>
---
 src/instructlab/sdg/utils/models.py | 1 +
 1 file changed, 1 insertion(+)

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",
 }