chenchenghao/sglang
1
0
Fork 0
Code Issues Pull Requests Actions 12 Packages Projects Releases Wiki Activity

Remove deepseek-r1 from THINKING_MODE_CHOICES in run_eval.py (#17178)

Browse Source
This commit is contained in:
hlu1
2026-01-15 16:53:06 -08:00
committed by GitHub
parent e3a95077bc
commit 0e86de7c0b
1 changed files with 3 additions and 2 deletions
Download Patch File Download Diff File Expand all files Collapse all files
python/sglang/test/run_eval.py
+3 -2
View File
@@ -22,6 +22,7 @@ def get_thinking_kwargs(args):
if thinking_mode == "deepseek-v3": if thinking_mode == "deepseek-v3":
thinking_param = "thinking" thinking_param = "thinking"
else: else:
# Qwen3
thinking_param = "enable_thinking" thinking_param = "enable_thinking"
return { return {
"chat_template_kwargs": {thinking_param: True}, "chat_template_kwargs": {thinking_param: True},
@@ -203,7 +204,7 @@ def run_eval(args):
return metrics return metrics
THINKING_MODE_CHOICES = ["deepseek-r1", "deepseek-v3", "qwen3"] THINKING_MODE_CHOICES = ["deepseek-v3", "qwen3"]
if __name__ == "__main__": if __name__ == "__main__":
parser = argparse.ArgumentParser() parser = argparse.ArgumentParser()
@@ -241,7 +242,7 @@ if __name__ == "__main__":
default=None, default=None,
type=str, type=str,
choices=THINKING_MODE_CHOICES, choices=THINKING_MODE_CHOICES,
help="Enable thinking mode in Deepseek R1, V3.1/3.2, or Qwen3", help="Enable thinking mode in Deepseek V3.1/3.2, or Qwen3.--reasoning-parser must be set when launching the server.",
) )
# LongBench-v2 specific arguments # LongBench-v2 specific arguments