From dc454fdc0ffde12041c71fda0fb324e731f55ae3 Mon Sep 17 00:00:00 2001 From: =?UTF-8?q?Cl=C3=A9mentine=20Fourrier?= <22726840+clefourrier@users.noreply.github.com> Date: Wed, 7 Feb 2024 14:54:17 +0100 Subject: [PATCH] Update src/lighteval/models/base_model.py Co-authored-by: Nathan Habib <30601243+NathanHB@users.noreply.github.com> --- src/lighteval/models/base_model.py | 2 +- 1 file changed, 1 insertion(+), 1 deletion(-) diff --git a/src/lighteval/models/base_model.py b/src/lighteval/models/base_model.py index aed6ca8e..6b67e4c0 100644 --- a/src/lighteval/models/base_model.py +++ b/src/lighteval/models/base_model.py @@ -626,7 +626,7 @@ def _loglikelihood_tokens( return dataset.get_original_order(res) def prepare_batch( - self, batch: list[str], padding_length: int, max_context: Optional[int] = None, single_token: bool = False + self, batch: list[Request], padding_length: int, max_context: Optional[int] = None, single_token: bool = False ): """Tokenize a batch of inputs and return also the length, truncations and padding""" if single_token: