Skip to content

GitLab

Commit 39a11d02 authored by daniel-furman's avatar daniel-furman
Browse files

llama test

parent 43dee065
Hide whitespace changes
Inline Side-by-side
Showing with 1 addition and 1 deletion
lm_eval/models/huggingface.py
View file @ 39a11d02
...@@ -710,7 +710,7 @@ class HFLM(LM): ...@@ -710,7 +710,7 @@ class HFLM(LM):
tokenize=False, tokenize=False,
add_generation_prompt=True, add_generation_prompt=True,
) )
req.args = (context, req.args[1]) req.args = (context, req.args[1].strip())
new_reqs.append(req) new_reqs.append(req)
return new_reqs return new_reqs
......
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