biondizzle/vllm
1
0
Fork 0
Code Issues Pull Requests Actions 2 Packages Projects Releases Wiki Activity

Revert "Refactor llama family models (#2637)" (#2851)

Browse Source 
This reverts commit 5c976a7e1a.
This commit is contained in:
Philipp Moritz
2024-02-13 09:24:59 -08:00
committed by GitHub
parent 5c976a7e1a
commit ea356004d4
17 changed files with 2720 additions and 236 deletions
Download Patch File Download Diff File Expand all files Collapse all files

8
vllm/transformers_utils/configs/__init__.py
View File

@@ -1,12 +1,20 @@
from vllm.transformers_utils.configs.aquila import AquilaConfig
from vllm.transformers_utils.configs.baichuan import BaiChuanConfig
from vllm.transformers_utils.configs.chatglm import ChatGLMConfig
from vllm.transformers_utils.configs.mpt import MPTConfig
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.
from vllm.transformers_utils.configs.falcon import RWConfig
from vllm.transformers_utils.configs.yi import YiConfig
__all__ = [
"AquilaConfig",
"BaiChuanConfig",
"ChatGLMConfig",
"MPTConfig",
"QWenConfig",
"RWConfig",
"YiConfig",
]