40 lines
920 B
Python
40 lines
920 B
Python
# SPDX-License-Identifier: Apache-2.0
|
|
# SPDX-FileCopyrightText: Copyright contributors to the vLLM project
|
|
|
|
from vllm.parser.abstract_parser import (
|
|
DelegatingParser,
|
|
Parser,
|
|
_WrappedParser,
|
|
)
|
|
from vllm.parser.parser_manager import ParserManager
|
|
|
|
__all__ = [
|
|
"Parser",
|
|
"DelegatingParser",
|
|
"ParserManager",
|
|
"_WrappedParser",
|
|
]
|
|
|
|
_PARSERS_TO_REGISTER = {
|
|
"minimax_m2": ( # name
|
|
"minimax_m2_parser", # filename
|
|
"MiniMaxM2Parser", # class_name
|
|
),
|
|
}
|
|
|
|
# Register lazy parsers
|
|
ParserManager.register_lazy_module(
|
|
name="minimax_m2",
|
|
module_path="vllm.parser.minimax_m2_parser",
|
|
class_name="MiniMaxM2Parser",
|
|
)
|
|
|
|
|
|
def register_lazy_parsers():
|
|
for name, (file_name, class_name) in _PARSERS_TO_REGISTER.items():
|
|
module_path = f"vllm.parser.{file_name}"
|
|
ParserManager.register_lazy_module(name, module_path, class_name)
|
|
|
|
|
|
register_lazy_parsers()
|