bugfix

游雁

2024-04-15 199f6e630af9d9ccfb8c50bcd1f7ba0392c9bd5e

 funasr/models/paraformer/decoder.py

@@ -351,9 +351,9 @@
        hlens: torch.Tensor,
        ys_in_pad: torch.Tensor,
        ys_in_lens: torch.Tensor,
        return_hidden: bool = False,
        return_both: bool= False,
        chunk_mask: torch.Tensor = None,
        return_hidden: bool = False,
        return_both: bool = False,
    ) -> Tuple[torch.Tensor, torch.Tensor]:
        """Forward decoder.