Skip to content
3 changes: 2 additions & 1 deletion src/lighteval/tasks/multilingual/tasks.py
Original file line number Diff line number Diff line change
Expand Up @@ -463,7 +463,8 @@
Language.TURKISH,
Language.VIETNAMESE,
Language.CHINESE,
# Optionally: Haitian, Quechu
Language.HAITIAN,
Language.QUECHUA,
]
for formulation in [MCFFormulation(), CFFormulation(), HybridFormulation()]
]
Expand Down
2 changes: 2 additions & 0 deletions src/lighteval/tasks/templates/nli.py
Original file line number Diff line number Diff line change
Expand Up @@ -244,6 +244,8 @@ def prompt_fn(line: dict, task_name: str):
choices_str = f"{translation_literals.comma}{translation_literals.word_space}".join(rearanged_labales[:-1])
hypothesis = f"{hypothesis.rstrip(PUNCT)}{translation_literals.sentence_space}{choices_str}{translation_literals.word_space}{translation_literals.or_word}{translation_literals.word_space}{rearanged_labales[-1]}{translation_literals.question_mark}"

# (hynky1999): Ideally we would not compute logprobs of the Yes/No/Also in CF fomulation. However as of right now lighteval doesn't allow to
# use multi-context.
row = {
"instruction": input_data.get("instruction", ""),
"premise": premise,
Expand Down
Loading
Loading