Support Yi model (#1567)

2023-11-07 07:26:03 +08:00
parent 8516999495
commit e7f579eb97
6 changed files with 496 additions and 0 deletions
--- a/vllm/transformers_utils/configs/init.py
+++ b/vllm/transformers_utils/configs/init.py
@@ -5,10 +5,12 @@ from vllm.transformers_utils.configs.qwen import QWenConfig
 # tiiuae/falcon-7b(-instruct) models. Newer Falcon models will use the
 # `FalconConfig` class from the official HuggingFace transformers library.
 from vllm.transformers_utils.configs.falcon import RWConfig
+from vllm.transformers_utils.configs.yi import YiConfig

 __all__ = [
    "BaiChuanConfig",
    "AquilaConfig",
    "QWenConfig",
    "RWConfig",
+    "YiConfig",
 ]