Skip to content

GitLab

Commit 842139ae authored by cjlovering's avatar cjlovering
Browse files

Merge branch 'master' of github.com:cjlovering/ps-eh

parents 23a6ee7f 61b57292
Hide whitespace changes
Inline Side-by-side
Showing with 1 addition and 1 deletion
lm_eval/models/gpt2.py
View file @ 842139ae
...@@ -139,7 +139,7 @@ class HFLM(BaseLM): ...@@ -139,7 +139,7 @@ class HFLM(BaseLM):
return transformers.StoppingCriteriaList([ return transformers.StoppingCriteriaList([
MultitokenEOSCriteria(stopping_criteria_ids, self.tokenizer), MultitokenEOSCriteria(stopping_criteria_ids, self.tokenizer),
EOSCriteria(stopping_criteria_ids) EOSCriteria(self.tokenizer.eos_token)
]) ])
def _model_generate(self, context, max_length, stopping_criteria_ids): def _model_generate(self, context, max_length, stopping_criteria_ids):
......
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