diff --git a/optimum/utils/normalized_config.py b/optimum/utils/normalized_config.py index a894001d35..682f70e3ca 100644 --- a/optimum/utils/normalized_config.py +++ b/optimum/utils/normalized_config.py @@ -253,6 +253,7 @@ class NormalizedConfigManager: "pix2struct": Pix2StructNormalizedTextConfig, "phi": NormalizedTextConfig, "phi3": NormalizedTextConfigWithGQA, + "phi3small": NormalizedTextConfigWithGQA, "poolformer": NormalizedVisionConfig, "regnet": NormalizedVisionConfig, "resnet": NormalizedVisionConfig,