"docs/vscode:/vscode.git/clone" did not exist on "fe2016de2d2d68bb967e9c43ee168db01322b0f1"
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
......@@ -36,8 +36,8 @@ class Ernie45ReasoningParser(BaseThinkingReasoningParser):
"""The token that ends reasoning content."""
return "</think>"
def __init__(self, tokenizer: PreTrainedTokenizerBase):
super().__init__(tokenizer)
def __init__(self, tokenizer: PreTrainedTokenizerBase, *args, **kwargs):
super().__init__(tokenizer, *args, **kwargs)
if not self.model_tokenizer:
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