add mistral support

Signed-off-by: changwangss <[email protected]>
huggingface · Oct 16, 2023 · 1bb19e1 · 1bb19e1
1 parent 10c7c53
commit 1bb19e1
Showing 1 changed file with 1 addition and 1 deletion.
diff --git 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