[hot-fix] Fix CI break which caused by adding thinking_mode in eval (#11192)
This commit is contained in:
@@ -17,13 +17,14 @@ from sglang.test.simple_eval_common import (
|
|||||||
|
|
||||||
|
|
||||||
def get_thinking_kwargs(args):
|
def get_thinking_kwargs(args):
|
||||||
if args.thinking_mode in THINKING_MODE_CHOICES:
|
thinking_mode = getattr(args, "thinking_mode", None)
|
||||||
thinking_param = (
|
if thinking_mode in THINKING_MODE_CHOICES:
|
||||||
"thinking" if args.thinking_mode == "deepseek-v3" else "enable_thinking"
|
if thinking_mode == "deepseek-v3":
|
||||||
)
|
thinking_param = "thinking"
|
||||||
|
else:
|
||||||
|
thinking_param = "enable_thinking"
|
||||||
return {
|
return {
|
||||||
"chat_template_kwargs": {thinking_param: True},
|
"chat_template_kwargs": {thinking_param: True},
|
||||||
"separate_reasoning": True,
|
|
||||||
}
|
}
|
||||||
return {}
|
return {}
|
||||||
|
|
||||||
|
|||||||
Reference in New Issue
Block a user