13 lines
466 B
Python
13 lines
466 B
Python
from vllm.transformers_utils.configs.mpt import MPTConfig
|
|
from vllm.transformers_utils.configs.baichuan import BaiChuanConfig
|
|
# 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
|
|
|
|
__all__ = [
|
|
"MPTConfig",
|
|
"BaiChuanConfig",
|
|
"RWConfig",
|
|
]
|