update funasr-runtime-sdk-gpu-0.1.1

雾聪

2024-07-01 31bf3a88a09f6e7c895224f4dc75e8f2c138d5c8

 funasr/models/sanm/attention.py

@@ -780,7 +780,7 @@
        return q, k, v

    def forward_attention(self, value, scores, mask, ret_attn):
        scores = scores + mask
        scores = scores + mask.to(scores.device)

        self.attn = torch.softmax(scores, dim=-1)
        context_layer = torch.matmul(self.attn, value)  # (batch, head, time1, d_k)