diff --git a/optimum/utils/normalized_config.py b/optimum/utils/normalized_config.py index b5fba44d543..7eda65cb47e 100644 --- a/optimum/utils/normalized_config.py +++ b/optimum/utils/normalized_config.py @@ -238,12 +238,14 @@ class NormalizedConfigManager: "marian": BartLikeNormalizedTextConfig, "mbart": BartLikeNormalizedTextConfig, "mistral": NormalizedTextConfigWithGQA, + "mixtral": NormalizedTextConfigWithGQA, "mt5": T5LikeNormalizedTextConfig, "m2m-100": BartLikeNormalizedTextConfig, "nystromformer": NormalizedTextConfig, "opt": NormalizedTextConfig, "pegasus": BartLikeNormalizedTextConfig, "pix2struct": Pix2StructNormalizedTextConfig, + "phi": NormalizedTextConfig, "poolformer": NormalizedVisionConfig, "regnet": NormalizedVisionConfig, "resnet": NormalizedVisionConfig,