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

This commit is contained in:
hlu1
2026-01-15 16:53:06 -08:00
committed by GitHub
parent e3a95077bc
commit 0e86de7c0b

View File

@@ -22,6 +22,7 @@ def get_thinking_kwargs(args):
if thinking_mode == "deepseek-v3":
thinking_param = "thinking"
else:
# Qwen3
thinking_param = "enable_thinking"
return {
"chat_template_kwargs": {thinking_param: True},
@@ -203,7 +204,7 @@ def run_eval(args):
return metrics
THINKING_MODE_CHOICES = ["deepseek-r1", "deepseek-v3", "qwen3"]
THINKING_MODE_CHOICES = ["deepseek-v3", "qwen3"]
if __name__ == "__main__":
parser = argparse.ArgumentParser()
@@ -241,7 +242,7 @@ if __name__ == "__main__":
default=None,
type=str,
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