__init__.py 787 Bytes
Newer Older
1
2
from vllm.transformers_utils.configs.aquila import AquilaConfig
from vllm.transformers_utils.configs.baichuan import BaiChuanConfig
GoHomeToMacDonal's avatar
GoHomeToMacDonal committed
3
from vllm.transformers_utils.configs.chatglm import ChatGLMConfig
4
from vllm.transformers_utils.configs.mpt import MPTConfig
5
from vllm.transformers_utils.configs.qwen import QWenConfig
Zhuohan Li's avatar
Zhuohan Li committed
6
7
8
9
# 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
10
from vllm.transformers_utils.configs.yi import YiConfig
11
12

__all__ = [
13
14
    "AquilaConfig",
    "BaiChuanConfig",
GoHomeToMacDonal's avatar
GoHomeToMacDonal committed
15
    "ChatGLMConfig",
16
    "MPTConfig",
17
    "QWenConfig",
Zhuohan Li's avatar
Zhuohan Li committed
18
    "RWConfig",
19
    "YiConfig",
20
]