Skip to content

GitLab

Unverified Commit 43a6acfb authored by CSWYF3634076's avatar CSWYF3634076 Committed by GitHub
Browse files

[Model] fix ernie45 reasoning_parser (#27973) 


Signed-off-by: default avatarwangyafeng <wangyafeng@baidu.com>
parent 58279c60
Show whitespace changes
Inline Side-by-side
Showing with 2 additions and 2 deletions
vllm/reasoning/ernie45_reasoning_parser.py
View file @ 43a6acfb
...@@ -36,8 +36,8 @@ class Ernie45ReasoningParser(BaseThinkingReasoningParser): ...@@ -36,8 +36,8 @@ class Ernie45ReasoningParser(BaseThinkingReasoningParser):
"""The token that ends reasoning content.""" """The token that ends reasoning content."""
return "</think>" return "</think>"
def __init__(self, tokenizer: PreTrainedTokenizerBase): def __init__(self, tokenizer: PreTrainedTokenizerBase, *args, **kwargs):
super().__init__(tokenizer) super().__init__(tokenizer, *args, **kwargs)
if not self.model_tokenizer: if not self.model_tokenizer:
raise ValueError( raise ValueError(
......
Markdown is supported
0% or .
You are about to add 0 people to the discussion. Proceed with caution.
Finish editing this message first!
Please register or to comment