Expand support (huggingface#1864)

add
kunal-vaishnavi · May 21, 2024 · e0f5812 · e0f5812
1 parent 02c6ed5
commit e0f5812
Showing 1 changed file with 1 addition and 0 deletions.
diff --git 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,