Shi Xian
2024-01-16 eba1fccfa07665ed9cada19cfab6cd342e03112e
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