Revert MptConfig to MPTConfig (#1668)

2023-11-16 01:19:39 -08:00
parent 7076fa1c9f
commit b514d3c496
6 changed files with 260 additions and 26 deletions
--- a/vllm/transformers_utils/config.py
+++ b/vllm/transformers_utils/config.py
@@ -1,6 +1,6 @@
 from typing import Optional

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

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

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