File tree Expand file tree Collapse file tree 1 file changed +5
-1
lines changed Expand file tree Collapse file tree 1 file changed +5
-1
lines changed Original file line number Diff line number Diff line change @@ -188,7 +188,11 @@ def run_codegen(
188
188
extra = "-" + subset if subset != "full" else ""
189
189
if reasoning_effort and model .startswith ("o1-" ) or model .startswith ("o3-" ) or model .endswith ("-reasoner" ):
190
190
model = model + f"--{ reasoning_effort } "
191
- identifier = model .replace ("/" , "--" ) + "--skip_prefill" if skip_prefill else "" + f"--{ revision } --bigcodebench{ extra } -{ split } --{ backend } -{ temperature } -{ n_samples } -sanitized_calibrated.jsonl"
191
+
192
+ if skip_prefill :
193
+ identifier = model .replace ("/" , "--" ) + "--skip_prefill" + f"--{ revision } --bigcodebench{ extra } -{ split } --{ backend } -{ temperature } -{ n_samples } -sanitized_calibrated.jsonl"
194
+ else :
195
+ identifier = model .replace ("/" , "--" ) + f"--{ revision } --bigcodebench{ extra } -{ split } --{ backend } -{ temperature } -{ n_samples } -sanitized_calibrated.jsonl"
192
196
193
197
target_path = os .path .join (root , identifier )
194
198
You can’t perform that action at this time.
0 commit comments