f96951464d4dc822d8a97052cdb765686084aacc,onmt/transforms/bart.py,BARTNoiseTransform,warm_up,#BARTNoiseTransform#Any#,346

Before Change

    def warm_up(self, vocabs):
        if vocabs is None:
            return
        self.vocab = vocabs["src"].itos

        subword_type = self.opts.src_subword_type
        if self.opts.mask_length == "subword":

After Change

                       "or N tokens. (use -1 for N)")

    def warm_up(self, vocabs):
        super().warm_up(None)
        if vocabs is None:
            return
        self.vocabs = vocabs

Instances