funasr/models/encoder/sanm_encoder.py
@@ -379,7 +379,10 @@ xs_pad = xs_pad else: xs_pad = self.embed(xs_pad, cache) xs_pad = self._add_overlap_chunk(xs_pad, cache) if cache["tail_chunk"]: 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) xs_pad, masks = encoder_outs[0], encoder_outs[1] intermediate_outs = []
