diff --git a/optimum/utils/normalized_config.py b/optimum/utils/normalized_config.py index 21470cb61f..b3dc3733d5 100644 --- a/optimum/utils/normalized_config.py +++ b/optimum/utils/normalized_config.py @@ -303,6 +303,7 @@ class NormalizedConfigManager: "segformer": NormalizedSegformerConfig, "speech_to_text": SpeechToTextLikeNormalizedTextConfig, "splinter": NormalizedTextConfig, + "swinv2": NormalizedVisionConfig, "t5": T5LikeNormalizedTextConfig, "trocr": TrOCRLikeNormalizedTextConfig, "vision-encoder-decoder": NormalizedEncoderDecoderConfig,