funasr1.0 ct-transformer streaming

游雁

2024-01-14 99730b35f47579eb99b5e4ba0e6ca99901c23955

 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: