志浩
2023-03-09 7335c2070965e8d0c84a7bc7d6fdae5f9a3910b5
funasr/models/encoder/sanm_encoder.py
@@ -293,7 +293,7 @@
            position embedded tensor and mask
        """
        masks = (~make_pad_mask(ilens)[:, None, :]).to(xs_pad.device)
        xs_pad *= self.output_size()**0.5
        xs_pad = xs_pad * self.output_size()**0.5
        if self.embed is None:
            xs_pad = xs_pad
        elif (