hnluo
2023-04-28 6013d3c4a9afa81f8f7bdca4e3d9fe84639eb39b
funasr/models/encoder/sanm_encoder.py
@@ -380,7 +380,7 @@
        else:
            xs_pad = self.embed(xs_pad, cache)
        if cache["tail_chunk"]:
            xs_pad = cache["feats"]
            xs_pad = to_device(cache["feats"], device=xs_pad.device)
        else:
            xs_pad = self._add_overlap_chunk(xs_pad, cache)
        encoder_outs = self.encoders0(xs_pad, None, None, None, None)