__init__.py 715 Bytes
Newer Older
1
from vllm.transformers_utils.configs.mpt import MPTConfig
codethazine's avatar
codethazine committed
2
from vllm.transformers_utils.configs.baichuan import BaiChuanConfig
shunxing1234's avatar
shunxing1234 committed
3
from vllm.transformers_utils.configs.aquila import AquilaConfig
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
Woosuk Kwon's avatar
Woosuk Kwon committed
9
from vllm.transformers_utils.configs.mistral import MistralConfig
10
11
12

__all__ = [
    "MPTConfig",
codethazine's avatar
codethazine committed
13
    "BaiChuanConfig",
shunxing1234's avatar
shunxing1234 committed
14
    "AquilaConfig",
Qing's avatar
Qing committed
15
    "QWenConfig",
Zhuohan Li's avatar
Zhuohan Li committed
16
    "RWConfig",
Woosuk Kwon's avatar
Woosuk Kwon committed
17
    "MistralConfig",
18
]