add models config

Signed-off-by: changwangss <[email protected]>
huggingface · Oct 27, 2023 · deda7e3 · deda7e3
1 parent 313e1bd
commit deda7e3
Showing 1 changed file with 5 additions and 0 deletions.
diff --git a/optimum/utils/normalized_config.py b/optimum/utils/normalized_config.py
@@ -254,6 +254,11 @@ class NormalizedConfigManager:
         "xlm-roberta": NormalizedTextConfig,
         "yolos": NormalizedVisionConfig,
         "mpt": MPTNormalizedTextConfig,
+        "mixformer-sequential": GPTBigCodeNormalizedTextConfig,
+        "baichuan": NormalizedTextConfig,
+        "qwen": NormalizedTextConfig,
+        "chatglm": NormalizedTextConfig.with_args(num_layers="num_layers"),
+
     }
 
     @classmethod