__init__.py 712 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
Qing's avatar
Qing committed
4
from vllm.transformers_utils.configs.qwen import QWenConfig
Zhuohan Li's avatar
Zhuohan Li committed
5
6
7
8
# 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
9
from vllm.transformers_utils.configs.yi import YiConfig
10
11

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