Skip to content

GitLab

"vscode:/vscode.git/clone" did not exist on "a7bf2b418e3c1fd38fefcedaff35c2c13dd293e6"
Unverified Commit 877e9a61 authored by Chris's avatar Chris Committed by GitHub
Browse files

Fix positional arguments in HF model generate

parent a3252ed7
Hide whitespace changes
Inline Side-by-side
Showing with 1 addition and 1 deletion
lm_eval/models/huggingface.py
View file @ 877e9a61
...@@ -508,7 +508,7 @@ class HFLM(LM): ...@@ -508,7 +508,7 @@ class HFLM(LM):
self.tokenizer, stop, 1, context.shape[0] self.tokenizer, stop, 1, context.shape[0]
) )
return self.model.generate( return self.model.generate(
context, input_ids=context,
max_length=max_length, max_length=max_length,
stopping_criteria=stopping_criteria, stopping_criteria=stopping_criteria,
pad_token_id=self.eot_token_id, pad_token_id=self.eot_token_id,
......
Markdown is supported
0% or .
You are about to add 0 people to the discussion. Proceed with caution.
Finish editing this message first!
Please register or to comment