From 2bc0b7fcd3826f2f9819ce07929573c4b5983f46 Mon Sep 17 00:00:00 2001 From: =?UTF-8?q?Cl=C3=A9mentine?= <cle.fourrier@gmail.com> Date: Mon, 9 Dec 2024 19:20:27 +0100 Subject: [PATCH] fix --- src/lighteval/models/endpoints/endpoint_model.py | 2 +- 1 file changed, 1 insertion(+), 1 deletion(-) diff --git a/src/lighteval/models/endpoints/endpoint_model.py b/src/lighteval/models/endpoints/endpoint_model.py index 9d4e4afa7..b34de9fe2 100644 --- a/src/lighteval/models/endpoints/endpoint_model.py +++ b/src/lighteval/models/endpoints/endpoint_model.py @@ -79,7 +79,7 @@ class InferenceModelConfig: model: str add_special_tokens: bool = True - generation_config: TextGenerationInputGenerateParameters + generation_config: dict = dict @dataclass