Commit e04aab6d authored by Jan-Christoph Kalo's avatar Jan-Christoph Kalo
Browse files

added task to init

parent 27a8f231
...@@ -5,6 +5,9 @@ from typing import List, Union ...@@ -5,6 +5,9 @@ from typing import List, Union
import sacrebleu import sacrebleu
import lm_eval.base import lm_eval.base
from . import bigsciencelama
from . import lama
from . import superglue from . import superglue
from . import glue from . import glue
from . import arc from . import arc
...@@ -308,6 +311,11 @@ TASK_REGISTRY = { ...@@ -308,6 +311,11 @@ TASK_REGISTRY = {
"gem_xsum_challenge_test_bfp_05": gem_xsum.GEMXSUMChallgeTestBFP05, "gem_xsum_challenge_test_bfp_05": gem_xsum.GEMXSUMChallgeTestBFP05,
"gem_xsum_challenge_test_nopunc": gem_xsum.GEMXSUMChallgeTestNopunc, "gem_xsum_challenge_test_nopunc": gem_xsum.GEMXSUMChallgeTestNopunc,
"gem_xsum_challenge_test_covid": gem_xsum.GEMXSUMChallgeTestCovid, "gem_xsum_challenge_test_covid": gem_xsum.GEMXSUMChallgeTestCovid,
#LAMA
"lama-trex":lama.Trex,
"bigscience-lama":bigsciencelama.BigScienceLAMA,
} }
......
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