Merge pull request #936 from alibaba-damo-academy/dev_lhn

Dev lhn
This commit is contained in:
hnluo 2023-09-13 10:54:55 +08:00 committed by GitHub
commit b091828cea
No known key found for this signature in database
GPG Key ID: 4AEE18F83AFDEB23

View File

@ -430,7 +430,7 @@ class Speech2TextParaformer:
# b. Forward Encoder
if decoding_ind is None:
decoding_ind = self.decoding_ind
decoding_ind = 0 if self.decoding_ind is None else self.decoding_ind
enc, enc_len = self.asr_model.encode(**batch, ind=decoding_ind)
if isinstance(enc, tuple):
enc = enc[0]