We read every piece of feedback, and take your input very seriously.
To see all available qualifiers, see our documentation.
There was an error while loading. Please reload this page.
1 parent a810f31 commit e8798f4Copy full SHA for e8798f4
bigcodebench/generate.py
@@ -135,7 +135,7 @@ def run_codegen(
135
strip_newlines: bool = False,
136
direct_completion: bool = False,
137
resume: bool = True,
138
- id_range: Tuple[int, int] = None,
+ id_range: str = None,
139
backend: str = "vllm",
140
base_url: str = None,
141
tp: int = 1,
@@ -152,6 +152,7 @@ def run_codegen(
152
print("Greedy decoding ON (--greedy): setting n_samples=1, temperature=0")
153
154
if id_range is not None:
155
+ id_range = [int(i) for i in id_range.split("-")]
156
assert len(id_range) == 2, "id_range must be a list of length 2"
157
assert id_range[0] < id_range[1], "id_range must be increasing"
158
id_range = tuple(id_range)
0 commit comments