__init__.py 625 Bytes
Newer Older
codethazine's avatar
codethazine committed
1
from vllm.transformers_utils.configs.baichuan import BaiChuanConfig
shunxing1234's avatar
shunxing1234 committed
2
from vllm.transformers_utils.configs.aquila import AquilaConfig
Qing's avatar
Qing committed
3
from vllm.transformers_utils.configs.qwen import QWenConfig
Zhuohan Li's avatar
Zhuohan Li committed
4
5
6
7
# 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
8
from vllm.transformers_utils.configs.yi import YiConfig
9
10

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