21 lines
787 B
Python
21 lines
787 B
Python
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",
|
|
]
|