Skip to content

Commit 33b12fb

Browse files
fix resource leakage
1 parent 71f1e3d commit 33b12fb

File tree

1 file changed

+1
-1
lines changed

1 file changed

+1
-1
lines changed

examples/verify_algo.py

Lines changed: 1 addition & 1 deletion
Original file line numberDiff line numberDiff line change
@@ -75,7 +75,7 @@ def verify_algos(
7575
with open("examples/amc23.jsonl", "r", encoding="utf-8") as f:
7676
requests = [json.loads(line) for line in f]
7777

78-
requests = requests[:2]
78+
requests = requests * trials
7979
prompts = [req["prompt"] for req in requests]
8080

8181
sampling_params = {"temperature": 0.6, "top_p": 0.95, "top_k": 20, "max_new_tokens": 8192}

0 commit comments

Comments
 (0)