From 1bb19e174212fbd185855507b3bfee3c3be13e46 Mon Sep 17 00:00:00 2001
From: changwangss <chang1.wang@intel.com>
Date: Sun, 15 Oct 2023 20:06:37 -0700
Subject: [PATCH] add mistral support

Signed-off-by: changwangss <chang1.wang@intel.com>
---
 optimum/utils/normalized_config.py | 2 +-
 1 file changed, 1 insertion(+), 1 deletion(-)

diff --git a/optimum/utils/normalized_config.py b/optimum/utils/normalized_config.py
index 4be6e41bbe2..1738bb73ec6 100644
--- a/optimum/utils/normalized_config.py
+++ b/optimum/utils/normalized_config.py
@@ -257,7 +257,7 @@ class NormalizedConfigManager:
         "yolos": NormalizedVisionConfig,
         "mpt": MPTNormalizedTextConfig,
         "gpt_bigcode": GPTBigCodeNormalizedTextConfig,
-        "mistral": MistralNormalizedTextConfig,
+        "mistral": MistralNormalizedTextConfig
     }
 
     @classmethod