From c133ccefffb8d3ce1c249dac60fd5ffe99f438be Mon Sep 17 00:00:00 2001
From: hnluo <haoneng.lhn@alibaba-inc.com>
Date: 星期四, 21 九月 2023 16:14:06 +0800
Subject: [PATCH] Merge pull request #974 from alibaba-damo-academy/dev_lhn
---
funasr/models/encoder/sanm_encoder.py | 4 ++--
1 files changed, 2 insertions(+), 2 deletions(-)
diff --git a/funasr/models/encoder/sanm_encoder.py b/funasr/models/encoder/sanm_encoder.py
index ac4240c..e04b9e7 100644
--- a/funasr/models/encoder/sanm_encoder.py
+++ b/funasr/models/encoder/sanm_encoder.py
@@ -945,11 +945,11 @@
for layer_idx, encoder_layer in enumerate(self.encoders):
encoder_outs = encoder_layer.forward_chunk(xs_pad, new_cache[layer_idx+len(self.encoders0)], cache["chunk_size"], cache["encoder_chunk_look_back"])
- xs_pad, new_cache[layer_idx+1] = encoder_outs[0], encoder_outs[1]
+ xs_pad, new_cache[layer_idx+len(self.encoders0)] = encoder_outs[0], encoder_outs[1]
if self.normalize_before:
xs_pad = self.after_norm(xs_pad)
- if cache["encoder_chunk_look_back"] > 0:
+ if cache["encoder_chunk_look_back"] > 0 or cache["encoder_chunk_look_back"] == -1:
cache["opt"] = new_cache
return xs_pad, ilens, None
--
Gitblit v1.9.1