Refactor llama family models (#2637)

2024-02-13 16:09:23 +08:00
parent f964493274
commit 5c976a7e1a
17 changed files with 236 additions and 2720 deletions
--- a/vllm/transformers_utils/config.py
+++ b/vllm/transformers_utils/config.py
@@ -5,14 +5,10 @@ from transformers import AutoConfig, PretrainedConfig
 from vllm.transformers_utils.configs import *

 _CONFIG_REGISTRY = {
-    "aquila": AquilaConfig,
-    "baichuan": BaiChuanConfig,
    "chatglm": ChatGLMConfig,
    "mpt": MPTConfig,
-    "qwen": QWenConfig,
    "RefinedWeb": RWConfig,  # For tiiuae/falcon-40b(-instruct)
    "RefinedWebModel": RWConfig,  # For tiiuae/falcon-7b(-instruct)
-    "yi": YiConfig,
 }