Skip to content

Commit 8cb50cc

Browse files
authored
Update task.py
1 parent 91230fe commit 8cb50cc

File tree

1 file changed

+1
-1
lines changed

1 file changed

+1
-1
lines changed

eval/task.py

Lines changed: 1 addition & 1 deletion
Original file line numberDiff line numberDiff line change
@@ -54,7 +54,7 @@ def _normalize_model_args(self, model: LM, instances: List[Instance]) -> List[In
5454
instance.args[1]["max_tokens"] = min(max_new_tokens, 16384)
5555
elif isinstance(model, lm_eval_models.vllm_causallms.VLLM):
5656
instance.args[1]["max_gen_toks"] = max_new_tokens
57-
else: # Huggingface does not support seed
57+
else: # Huggingface
5858
instance.args[1]["max_new_tokens"] = max_new_tokens
5959
return instances
6060

0 commit comments

Comments
 (0)