add QWen-7b (#685)

Co-authored-by: wq.chu <wq.chu@tianrang-inc.com>
This commit is contained in:
Qing
2023-08-09 04:50:38 +08:00
committed by GitHub
parent 79af7e96a0
commit a57d13cc96
6 changed files with 396 additions and 11 deletions

View File

@@ -1,5 +1,6 @@
from vllm.transformers_utils.configs.mpt import MPTConfig
from vllm.transformers_utils.configs.baichuan import BaiChuanConfig
from vllm.transformers_utils.configs.qwen import QWenConfig
# RWConfig is for the original tiiuae/falcon-40b(-instruct) and
# tiiuae/falcon-7b(-instruct) models. Newer Falcon models will use the
# `FalconConfig` class from the official HuggingFace transformers library.
@@ -8,5 +9,6 @@ from vllm.transformers_utils.configs.falcon import RWConfig
__all__ = [
"MPTConfig",
"BaiChuanConfig",
"QWenConfig",
"RWConfig",
]