Files
vllm/vllm/transformers_utils/configs/__init__.py
2024-02-13 00:09:23 -08:00

13 lines
463 B
Python

from vllm.transformers_utils.configs.chatglm import ChatGLMConfig
from vllm.transformers_utils.configs.mpt import MPTConfig
# 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__ = [
"ChatGLMConfig",
"MPTConfig",
"RWConfig",
]