Remove MPTConfig (#1529)

2023-11-01 15:29:05 -07:00
parent 7e90a2d117
commit 1fe0990023
6 changed files with 26 additions and 102 deletions
--- a/vllm/transformers_utils/config.py
+++ b/vllm/transformers_utils/config.py
@@ -1,11 +1,11 @@
 from typing import Optional

-from transformers import AutoConfig, PretrainedConfig
+from transformers import AutoConfig, MptConfig, PretrainedConfig

 from vllm.transformers_utils.configs import *  # pylint: disable=wildcard-import

 _CONFIG_REGISTRY = {
-    "mpt": MPTConfig,
+    "mpt": MptConfig,
    "baichuan": BaiChuanConfig,
    "aquila": AquilaConfig,
    "qwen": QWenConfig,