游雁
2024-01-15 831c48a886a5b879b46fc422cdc9c0898f6c34ec
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: