Unverified Commit f7881a27 authored by Liu Shaohui's avatar Liu Shaohui Committed by GitHub
Browse files

Add reasoning_effort param in TiktokenTokenizer.apply_chat_template (#9630)


Co-authored-by: default avatarShaohui Liu <liushaohui3@xiaomi.com>
Co-authored-by: default avatarXinyuan Tong <115166877+JustinTong0323@users.noreply.github.com>
parent fdff3167
......@@ -121,7 +121,12 @@ class TiktokenTokenizer:
return self.tokenizer.decode_batch(batch)
def apply_chat_template(
self, messages, tokenize, add_generation_prompt, tools=None
self,
messages,
tokenize,
add_generation_prompt,
tools=None,
reasoning_effort=None,
):
ret = self.chat_template_jinja.render(
messages=messages, add_generation_prompt=add_generation_prompt
......
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