From 4d18f77c8fed2b93357d7c6fd3507134bcd6a07d Mon Sep 17 00:00:00 2001 From: changwangss Date: Thu, 19 Oct 2023 15:33:53 -0700 Subject: [PATCH] support qwen Signed-off-by: changwangss --- optimum/utils/normalized_config.py | 1 + 1 file changed, 1 insertion(+) diff --git a/optimum/utils/normalized_config.py b/optimum/utils/normalized_config.py index 335bb4dabc..f9b337fb6c 100644 --- a/optimum/utils/normalized_config.py +++ b/optimum/utils/normalized_config.py @@ -244,6 +244,7 @@ class NormalizedConfigManager: "pegasus": BartLikeNormalizedTextConfig, "pix2struct": Pix2StructNormalizedTextConfig, "poolformer": NormalizedVisionConfig, + "qwen": NormalizedTextConfig, "regnet": NormalizedVisionConfig, "resnet": NormalizedVisionConfig, "roberta": NormalizedTextConfig,