From 90b1557996111abb055d2779771b293d7c4127c0 Mon Sep 17 00:00:00 2001 From: =?UTF-8?q?=E6=B8=B8=E9=9B=81?= Date: Mon, 22 Jul 2024 13:49:12 +0800 Subject: [PATCH] update --- 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 d464f1c70..6c6d15600 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(x.device) + padding_mask = (~make_pad_mask(olens)[:, None, :]).to(torch.bool).to(x.device) else: padding_mask = None