pull/2640/head
zxcd 3 years ago
parent dc009521df
commit 2471bffd8f

@ -1191,8 +1191,9 @@ class DecodingTask:
DecodingResult( DecodingResult(
audio_features=features, audio_features=features,
language=language, language=language,
language_probs=probs) for features, language, probs in language_probs=probs)
zip(audio_features, languages, language_probs) for features, language, probs in zip(audio_features, languages,
language_probs)
] ]
# repeat the audio & text tensors by the group size, for beam search or best-of-n sampling # repeat the audio & text tensors by the group size, for beam search or best-of-n sampling

Loading…
Cancel
Save