From 95b7d564a24ff5252057d94a54d56f4bf9888489 Mon Sep 17 00:00:00 2001 From: kxz2002 Date: Tue, 4 Nov 2025 13:07:57 +0800 Subject: [PATCH] fix parser register name --- fastdeploy/model_executor/models/ernie4_5_moe.py | 2 +- fastdeploy/model_executor/models/ernie4_5_mtp.py | 2 +- fastdeploy/reasoning/ernie_45_vl_thinking_reasoning_parser.py | 4 ++-- tests/entrypoints/openai/test_max_streaming_tokens.py | 2 +- 4 files changed, 5 insertions(+), 5 deletions(-) diff --git a/fastdeploy/model_executor/models/ernie4_5_moe.py b/fastdeploy/model_executor/models/ernie4_5_moe.py index c2baeb91049..e771d497a50 100644 --- a/fastdeploy/model_executor/models/ernie4_5_moe.py +++ b/fastdeploy/model_executor/models/ernie4_5_moe.py @@ -795,7 +795,7 @@ def _get_tensor_parallel_mappings(cls, config: PretrainedConfig, is_split=True): """ get_tensor_parallel_mappings """ - logger.info("erine inference model _get_tensor_parallel_mappings") + logger.info("ernie inference model _get_tensor_parallel_mappings") from fastdeploy.model_executor.models.tp_utils import ( build_expanded_keys, has_prefix, diff --git a/fastdeploy/model_executor/models/ernie4_5_mtp.py b/fastdeploy/model_executor/models/ernie4_5_mtp.py index fa8bb8e1085..7944f9f08dd 100644 --- a/fastdeploy/model_executor/models/ernie4_5_mtp.py +++ b/fastdeploy/model_executor/models/ernie4_5_mtp.py @@ -63,7 +63,7 @@ def _get_tensor_parallel_mappings(cls, config, is_split=True): """ get_tensor_parallel_mappings """ - logger.info("erine inference model _get_tensor_parallel_mappings") + logger.info("ernie inference model _get_tensor_parallel_mappings") from paddleformers.transformers.conversion_utils import split_or_merge_func diff --git a/fastdeploy/reasoning/ernie_45_vl_thinking_reasoning_parser.py b/fastdeploy/reasoning/ernie_45_vl_thinking_reasoning_parser.py index 72b045a3dc0..939a0a4348b 100644 --- a/fastdeploy/reasoning/ernie_45_vl_thinking_reasoning_parser.py +++ b/fastdeploy/reasoning/ernie_45_vl_thinking_reasoning_parser.py @@ -21,10 +21,10 @@ from fastdeploy.reasoning import ReasoningParser, ReasoningParserManager -@ReasoningParserManager.register_module("erine-45-vl-thinking") +@ReasoningParserManager.register_module("ernie-45-vl-thinking") class Ernie45VLThinkingReasoningParser(ReasoningParser): """ - Reasoning parser for ernir_vl model. + Reasoning parser for ernie_vl model. The ernie_vl model uses ...... tokens to denote reasoning text within its output. The model provides a strict switch to disable reasoning diff --git a/tests/entrypoints/openai/test_max_streaming_tokens.py b/tests/entrypoints/openai/test_max_streaming_tokens.py index 3a772f9193e..698ff368b95 100644 --- a/tests/entrypoints/openai/test_max_streaming_tokens.py +++ b/tests/entrypoints/openai/test_max_streaming_tokens.py @@ -532,7 +532,7 @@ async def mock_process_response_chat(response, stream, enable_thinking, include_ mock_processor_instance.process_response_chat = mock_process_response_chat mock_processor_instance.enable_multimodal_content = Mock(return_value=False) - mock_processor_instance.reasoning_parser = Mock(__class__.__name__ == "ErineTestReasoningParser") + mock_processor_instance.reasoning_parser = Mock(__class__.__name__ == "Ernie45VLThinkingReasoningParser") mock_processor_instance.data_processor = Mock( process_response_dict=lambda resp, stream, enable_thinking, include_stop_str_in_output: resp )