Support latest fairseq

2022-12-15 03:44:15 -08:00 · 2022-12-15 03:44:15 -08:00 · 7e12b582e4
commit 7e12b582e4
parent 2518ea030c
1 changed files with 1 additions and 0 deletions
--- a/examples/fairseq/tasks/pretraining.py
+++ b/examples/fairseq/tasks/pretraining.py
@ -180,6 +180,7 @@ class PLMTask(FairseqTask):
        epoch=1,
        data_buffer_size=0,
        disable_iterator_cache=False,
+        **kwargs,
    ):
        return MLMLoader(
            self.cfg,