From 609c0e7e0dc494992d68b74dc5bdff8d5e0d63a0 Mon Sep 17 00:00:00 2001 From: =?UTF-8?q?=E6=B8=B8=E9=9B=81?= Date: Wed, 24 Jul 2024 00:40:52 +0800 Subject: [PATCH] sdpa bugfix --- funasr/models/sense_voice/encoder.py | 2 +- 1 file changed, 1 insertion(+), 1 deletion(-) diff --git a/funasr/models/sense_voice/encoder.py b/funasr/models/sense_voice/encoder.py index 6c6d15600..64156e9dd 100644 --- a/funasr/models/sense_voice/encoder.py +++ b/funasr/models/sense_voice/encoder.py @@ -42,7 +42,7 @@ def sense_voice_encode_forward( olens = None if use_padmask and olens is not None: - padding_mask = (~make_pad_mask(olens)[:, None, :]).to(torch.bool).to(x.device) + padding_mask = (~make_pad_mask(olens)[:, :, None]).to(torch.bool).to(x.device) else: padding_mask = None