Unverified Commit a92d47eb authored by sdtblck's avatar sdtblck Committed by GitHub
Browse files

Update __init__.py

parent 4cd4b05c
...@@ -40,6 +40,7 @@ from . import cbt ...@@ -40,6 +40,7 @@ from . import cbt
from . import lambada_cloze from . import lambada_cloze
from . import pile from . import pile
from . import wikitext from . import wikitext
from . import lambada_multilingual
######################################## ########################################
# Translation tasks # Translation tasks
...@@ -97,6 +98,11 @@ TASK_REGISTRY = { ...@@ -97,6 +98,11 @@ TASK_REGISTRY = {
"drop": drop.DROP, "drop": drop.DROP,
"lambada": lambada.LAMBADA, "lambada": lambada.LAMBADA,
"lambada_cloze": lambada_cloze.LAMBADA_cloze, "lambada_cloze": lambada_cloze.LAMBADA_cloze,
# multlingual lambada
**multilingual_lambada.construct_tasks(),
lambada_multilingual
"wikitext": wikitext.WikiText, "wikitext": wikitext.WikiText,
# "cbt-cn": cbt.CBTCN, # disabled pending context length fix # "cbt-cn": cbt.CBTCN, # disabled pending context length fix
# "cbt-ne": cbt.CBTNE, # disabled pending context length fix # "cbt-ne": cbt.CBTNE, # disabled pending context length fix
...@@ -200,6 +206,7 @@ TASK_REGISTRY = { ...@@ -200,6 +206,7 @@ TASK_REGISTRY = {
"pile_ubuntu-irc": pile.PileUbuntuIrc, "pile_ubuntu-irc": pile.PileUbuntuIrc,
"pile_wikipedia": pile.PileWikipedia, "pile_wikipedia": pile.PileWikipedia,
"pile_youtubesubtitles": pile.PileYoutubeSubtitles, "pile_youtubesubtitles": pile.PileYoutubeSubtitles,
} }
......
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