We read every piece of feedback, and take your input very seriously.
To see all available qualifiers, see our documentation.
1 parent 23ad476 commit c3e02eaCopy full SHA for c3e02ea
src/lighteval/models/endpoints/openai_model.py
@@ -253,7 +253,7 @@ def greedy_until(
253
position=0,
254
disable=False, # self.disable_tqdm,
255
):
256
- max_new_tokens = 500 # dataset[0].generation_size # could be none
+ max_new_tokens = dataset[0].generation_size # could be none
257
return_logits = dataset[0].use_logits
258
num_samples = dataset[0].num_samples
259
contexts = [c.context for c in dataset]
0 commit comments