Skip to content

Commit d37847d

Browse files
committed
fix: customize lora output file
1 parent 82fc40d commit d37847d

File tree

1 file changed

+4
-1
lines changed

1 file changed

+4
-1
lines changed

bigcodebench/generate.py

Lines changed: 4 additions & 1 deletion
Original file line numberDiff line numberDiff line change
@@ -197,9 +197,12 @@ def run_codegen(
197197
if backend == "openai" and reasoning_effort and model.startswith("o1-") or model.startswith("o3-") or model.endswith("-reasoner"):
198198
model = model + f"--{reasoning_effort}"
199199

200+
if lora_path:
201+
model = model + f"--lora-{lora_path}"
202+
200203
if backend == "anthropic" and reasoning_budget and reasoning_beta:
201204
model = model + f"--{reasoning_budget}-{reasoning_beta}"
202-
205+
203206
if skip_prefill:
204207
identifier = model.replace("/", "--") + "--skip_prefill" + f"--{revision}--bigcodebench{extra}-{split}--{backend}-{temperature}-{n_samples}-sanitized_calibrated.jsonl"
205208
else:

0 commit comments

Comments
 (0)