游雁
2024-01-14 8912e0696af069de47646fdb8a9d9c4e086e88b3
funasr/models/ct_transformer/model.py
@@ -60,7 +60,7 @@
        
        
    def punc_forward(self, text: torch.Tensor, text_lengths: torch.Tensor) -> Tuple[torch.Tensor, None]:
    def punc_forward(self, text: torch.Tensor, text_lengths: torch.Tensor, **kwargs):
        """Compute loss value from buffer sequences.
        Args: