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