__init__.py 787 Bytes
Newer Older
shunxing1234's avatar
shunxing1234 committed
1
from vllm.transformers_utils.configs.aquila import AquilaConfig
GoHomeToMacDonal's avatar
GoHomeToMacDonal committed
2
3
from vllm.transformers_utils.configs.baichuan import BaiChuanConfig
from vllm.transformers_utils.configs.chatglm import ChatGLMConfig
4
from vllm.transformers_utils.configs.mpt import MPTConfig
Qing's avatar
Qing committed
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
Roy's avatar
Roy committed
10
from vllm.transformers_utils.configs.yi import YiConfig
11
12

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