Przeglądaj źródła

Update sanm_encoder.py

hnluo 2 lat temu
rodzic
commit
9ff5b683db
1 zmienionych plików z 4 dodań i 1 usunięć
  1. 4 1
      funasr/models/encoder/sanm_encoder.py

+ 4 - 1
funasr/models/encoder/sanm_encoder.py

@@ -379,7 +379,10 @@ class SANMEncoder(AbsEncoder):
             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 = cache["feats"]
+        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 = []