funasr/models/e2e_uni_asr.py
@@ -40,7 +40,7 @@ class UniASR(AbsESPnetModel): """ Author: Speech Lab, Alibaba Group, China Author: Speech Lab of DAMO Academy, Alibaba Group """ def __init__( @@ -206,7 +206,7 @@ with torch.no_grad(): speech_raw, encoder_out, encoder_out_lens = self.encode(speech, speech_lengths, ind=ind) else: speech_raw, encoder_out_lens = self.encode(speech, speech_lengths, ind=ind) speech_raw, encoder_out, encoder_out_lens = self.encode(speech, speech_lengths, ind=ind) intermediate_outs = None if isinstance(encoder_out, tuple):
