Skip to content

GitLab

Commit d362dfee authored by haileyschoelkopf's avatar haileyschoelkopf
Browse files

pass batch size

parent 5f1d18d4
Show whitespace changes
Inline Side-by-side
Showing with 1 addition and 0 deletions
lm_eval/evaluator.py
View file @ d362dfee
...@@ -79,6 +79,7 @@ def simple_evaluate( ...@@ -79,6 +79,7 @@ def simple_evaluate(
elif isinstance(model, transformers.PreTrainedModel): elif isinstance(model, transformers.PreTrainedModel):
lm = HFLM( lm = HFLM(
pretrained=model, pretrained=model,
batch_size=batch_size,
) )
no_cache = True no_cache = True
else: else:
......
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