shixian.shi
2024-01-15 296d73f7d647f63032380d19ad4f4e261f995c55
funasr/models/scama/sanm_encoder.py
@@ -423,7 +423,9 @@
                      xs_pad: torch.Tensor,
                      ilens: torch.Tensor,
                      cache: dict = None,
                      **kwargs,
                      ):
        is_final = kwargs.get("is_final", False)
        xs_pad *= self.output_size() ** 0.5
        if self.embed is None:
            xs_pad = xs_pad