d37529ed234ea9173ed35f6797a51a85378ecfca,fairseq/models/fairseq_incremental_decoder.py,FairseqIncrementalDecoder,reorder_incremental_state,#FairseqIncrementalDecoder#Any#Any#,62

Before Change

        previous time step. A typical use case is beam search, where the input
        order changes between time steps based on the selection of beams.
        
        seen = set()
        for module in self.modules():
            if (
                module != self
                and hasattr(module, "reorder_incremental_state")

After Change

        
        seen: Dict[int, Optional[Tensor]] = {}
        for _, module in self.named_modules():
            if hasattr(module, "reorder_incremental_state"):
                if id(module) not in seen and module is not self:
                    seen[id(module)] = None
                    result = module.reorder_incremental_state(incremental_state, new_order)
                    if result is not None:
                        incremental_state = result

    def set_beam_size(self, beam_size):
        Sets the beam size in the decoder and all children.
        if getattr(self, "_beam_size", -1) != beam_size:
            seen = set()

Instances