fix

huggingface · Dec 9, 2024 · 2bc0b7f · 2bc0b7f
1 parent dc90c92
commit 2bc0b7f
Showing 1 changed file with 1 addition and 1 deletion.
diff --git 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