__init__.py 553 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
8
9

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