pass batch size

d362dfee · haileyschoelkopf · 5f1d18d4 · d362dfee
Commit d362dfee authored Jun 27, 2023 by haileyschoelkopf
Hide whitespace changes
Inline Side-by-side

Showing with 1 addition and 0 deletions

lm_eval/evaluator.py lm_eval/evaluator.py +1 -0

No files found.
--- a/lm_eval/evaluator.py
+++ b/lm_eval/evaluator.py
@@ -79,6 +79,7 @@ def simple_evaluate(
    elif isinstance(model, transformers.PreTrainedModel):
        lm = HFLM(
                pretrained=model,
+                batch_size=batch_size,
                )
        no_cache = True
    else: