Migrate AquilaForCausalLM to LlamaForCausalLM (#2867)

2024-02-15 04:30:24 +08:00
parent 87069ccf68
commit 4efbac6d35
5 changed files with 2 additions and 417 deletions
--- a/vllm/transformers_utils/config.py
+++ b/vllm/transformers_utils/config.py
@@ -5,7 +5,6 @@ from transformers import AutoConfig, PretrainedConfig
 from vllm.transformers_utils.configs import *

 _CONFIG_REGISTRY = {
-    "aquila": AquilaConfig,
    "baichuan": BaiChuanConfig,
    "chatglm": ChatGLMConfig,
    "mpt": MPTConfig,