Skip to content

Commit c3e02ea

Browse files
committed
revert harcoding
1 parent 23ad476 commit c3e02ea

File tree

1 file changed

+1
-1
lines changed

1 file changed

+1
-1
lines changed

src/lighteval/models/endpoints/openai_model.py

Lines changed: 1 addition & 1 deletion
Original file line numberDiff line numberDiff line change
@@ -253,7 +253,7 @@ def greedy_until(
253253
position=0,
254254
disable=False, # self.disable_tqdm,
255255
):
256-
max_new_tokens = 500 # dataset[0].generation_size # could be none
256+
max_new_tokens = dataset[0].generation_size # could be none
257257
return_logits = dataset[0].use_logits
258258
num_samples = dataset[0].num_samples
259259
contexts = [c.context for c in dataset]

0 commit comments

Comments
 (0)