Update streaming_decode.py

This commit is contained in:
jinzr 2024-03-04 14:23:31 +08:00
parent 3d1a9680a8
commit d362eefa5c

View File

@ -377,11 +377,7 @@ def streaming_forward(
Returns encoder outputs, output lengths, and updated states.
"""
cached_embed_left_pad = states[-2]
(
x,
x_lens,
new_cached_embed_left_pad,
) = model.encoder_embed.streaming_forward(
(x, x_lens, new_cached_embed_left_pad,) = model.encoder_embed.streaming_forward(
x=features,
x_lens=feature_lens,
cached_left_pad=cached_embed_left_pad,