funasr/export/models/encoder/sanm_encoder.py
@@ -165,7 +165,7 @@ def prepare_mask(self, mask): mask_3d_btd = mask[:, :, None] sub_masks = subsequent_mask(mask.size(-1)) sub_masks = subsequent_mask(mask.size(-1)).type(torch.float32) if len(mask.shape) == 2: mask_4d_bhlt = 1 - sub_masks[:, None, None, :] elif len(mask.shape) == 3:
