From 7e12b582e4481b5ac2db8a02f2936891bbb7bf86 Mon Sep 17 00:00:00 2001 From: shumingma Date: Thu, 15 Dec 2022 03:44:15 -0800 Subject: [PATCH] Support latest fairseq --- examples/fairseq/tasks/pretraining.py | 1 + 1 file changed, 1 insertion(+) diff --git a/examples/fairseq/tasks/pretraining.py b/examples/fairseq/tasks/pretraining.py index 57b0628..2d32127 100644 --- 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,