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