paraformer onnx fp16导出方案 (#2264)

will_wang

2024-12-04 0c3c9be2c4c1c4e4da4628c3987708c9a0763391

 funasr/models/sond/encoder/ci_scorers.py

@@ -7,9 +7,9 @@
        super().__init__()

    def forward(
            self,
            xs_pad: torch.Tensor,
            spk_emb: torch.Tensor,
        self,
        xs_pad: torch.Tensor,
        spk_emb: torch.Tensor,
    ):
        # xs_pad: B, T, D
        # spk_emb: B, N, D
@@ -22,9 +22,9 @@
        super().__init__()

    def forward(
            self,
            xs_pad: torch.Tensor,
            spk_emb: torch.Tensor,
        self,
        xs_pad: torch.Tensor,
        spk_emb: torch.Tensor,
    ):
        # xs_pad: B, T, D
        # spk_emb: B, N, D