Commit 33ccde86 authored by Jonathan Tow's avatar Jonathan Tow
Browse files

Add `truthfulqa` back to the task registry

parent 36edc465
......@@ -43,7 +43,7 @@ from . import pile
from . import wikitext
from . import lambada_multilingual
from . import mutual
# from . import truthfulqa
from . import truthfulqa
########################################
# Translation tasks
......@@ -148,8 +148,8 @@ TASK_REGISTRY = {
"ethics_utilitarianism": hendrycks_ethics.EthicsUtilitarianism,
"ethics_virtue": hendrycks_ethics.EthicsVirtue,
# "truthfulqa_mc": truthfulqa.TruthfulQAMultipleChoice,
# "truthfulqa_gen": truthfulqa.TruthfulQAGeneration,
"truthfulqa_mc": truthfulqa.TruthfulQAMultipleChoice,
"truthfulqa_gen": truthfulqa.TruthfulQAGeneration,
# dialogue
"mutual": mutual.MuTual,
......
Markdown is supported
0% or .
You are about to add 0 people to the discussion. Proceed with caution.
Finish editing this message first!
Please register or to comment