Skip to content

Commit b0037bc

Browse files
authored
feat: add --disable-thinking option replaceing --enable-thinking (#86)
1 parent bae3ed2 commit b0037bc

File tree

1 file changed

+4
-4
lines changed

1 file changed

+4
-4
lines changed

run-openai.py

Lines changed: 4 additions & 4 deletions
Original file line numberDiff line numberDiff line change
@@ -89,16 +89,16 @@ def callback(
8989
parser.add_argument("--top-p", type=float, default=0.98, help="Top-p for sampling.")
9090
parser.add_argument("--extra-eos-tokens", type=str, nargs="+", help="Extra EOS strings")
9191
parser.add_argument(
92-
"--enable-thinking",
92+
"--disable-thinking",
9393
action="store_true",
94-
help="Enable reasoning when generation",
94+
help="Disable reasoning when generation by Qwen3 models",
9595
)
9696
args = parser.parse_args()
9797

9898
wrapped_callback = partial(
9999
callback,
100100
extra_eos_tokens=args.extra_eos_tokens,
101-
add_no_think=not args.enable_thinking,
101+
add_no_think=args.disable_thinking,
102102
)
103103

104104
pfgen.run_tasks(
@@ -109,5 +109,5 @@ def callback(
109109
temperature=args.temperature,
110110
top_p=args.top_p,
111111
num_trials=args.num_trials,
112-
enable_thinking=args.enable_thinking,
112+
enable_thinking=not args.disable_thinking,
113113
)

0 commit comments

Comments
 (0)