diff --git a/src/lighteval/tasks/lighteval_task.py b/src/lighteval/tasks/lighteval_task.py index 83723410f..25a6ee698 100644 --- a/src/lighteval/tasks/lighteval_task.py +++ b/src/lighteval/tasks/lighteval_task.py @@ -388,7 +388,7 @@ def get_docs(self, max_samples: int | None = None) -> list[Doc]: ) doc.sampling_methods.extend(self.sampling_methods) doc.generation_size = self.generation_size - doc.use_logits = True + doc.use_logits = doc.use_logits if doc.use_logits is not None else True doc.stop_sequences = self.stop_sequence doc.num_samples = max(self.num_samples) docs.append(doc)