Files
vllm/vllm/transformers_utils/configs/__init__.py
2023-11-01 15:29:05 -07:00

15 lines
553 B
Python

from vllm.transformers_utils.configs.baichuan import BaiChuanConfig
from vllm.transformers_utils.configs.aquila import AquilaConfig
from vllm.transformers_utils.configs.qwen import QWenConfig
# 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__ = [
"BaiChuanConfig",
"AquilaConfig",
"QWenConfig",
"RWConfig",
]