zhifu gao
2024-03-11 e847f85a14b85a94cfa6e109ecab69974eeeb443
funasr/models/ct_transformer_streaming/model.py
@@ -186,7 +186,7 @@
    
        return self
    def _export_forward(self, inputs: torch.Tensor,
    def export_forward(self, inputs: torch.Tensor,
                text_lengths: torch.Tensor,
                vad_indexes: torch.Tensor,
                sub_masks: torch.Tensor,