From 141cf8e8bf40f6316b8cf86fca964e73eb54cadd Mon Sep 17 00:00:00 2001 From: =?UTF-8?q?=E4=B9=9D=E8=80=B3?= Date: Thu, 30 Mar 2023 20:30:59 +0800 Subject: [PATCH] fix --- funasr/export/models/encoder/sanm_encoder.py | 3 ++- 1 file changed, 2 insertions(+), 1 deletion(-) diff --git a/funasr/export/models/encoder/sanm_encoder.py b/funasr/export/models/encoder/sanm_encoder.py index 8390f6822..44a48ff6a 100644 --- a/funasr/export/models/encoder/sanm_encoder.py +++ b/funasr/export/models/encoder/sanm_encoder.py @@ -164,6 +164,7 @@ class SANMVadEncoder(nn.Module): speech = speech * self._output_size ** 0.5 mask = self.make_pad_mask(speech_lengths) mask = self.prepare_mask(mask, sub_masks) + vad_mask = self.prepare_mask(mask, vad_mask) if self.embed is None: xs_pad = speech else: @@ -175,7 +176,7 @@ class SANMVadEncoder(nn.Module): # encoder_outs = self.model.encoders(xs_pad, mask) for layer_idx, encoder_layer in enumerate(self.model.encoders): if layer_idx == len(self.model.encoders) - 1: - mask = (mask[0], vad_mask) + mask = vad_mask encoder_outs = encoder_layer(xs_pad, mask) xs_pad, masks = encoder_outs[0], encoder_outs[1]