b12518d3cf
Signed-off-by: Thomas Parnell <tpa@zurich.ibm.com> Co-authored-by: Thomas Parnell <tpa@zurich.ibm.com> Co-authored-by: Nick Hill <nickhill@us.ibm.com> Co-authored-by: Davis Wertheimer <Davis.Wertheimer@ibm.com>
19 lines
725 B
Python
19 lines
725 B
Python
from vllm.transformers_utils.configs.chatglm import ChatGLMConfig
|
|
from vllm.transformers_utils.configs.dbrx import DbrxConfig
|
|
# 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
|
|
from vllm.transformers_utils.configs.jais import JAISConfig
|
|
from vllm.transformers_utils.configs.mlp_speculator import MLPSpeculatorConfig
|
|
from vllm.transformers_utils.configs.mpt import MPTConfig
|
|
|
|
__all__ = [
|
|
"ChatGLMConfig",
|
|
"DbrxConfig",
|
|
"MPTConfig",
|
|
"RWConfig",
|
|
"JAISConfig",
|
|
"MLPSpeculatorConfig",
|
|
]
|