Skip to content

Commit

Permalink
add ensemble decoding fix (#8427) (#8433)
Browse files Browse the repository at this point in the history
Signed-off-by: Nithin Rao Koluguri <nithinraok>
Co-authored-by: Nithin Rao <nithinrao.koluguri@gmail.com>
Signed-off-by: ataghibakhsh <ataghibakhsh@nvidia.com>
2 people authored and JRD971000 committed Feb 16, 2024

Verified

This commit was created on GitHub.com and signed with GitHub’s verified signature. The key has expired.
1 parent f2ea360 commit 370477b
Showing 1 changed file with 1 addition and 1 deletion.
2 changes: 1 addition & 1 deletion examples/asr/transcribe_speech.py
Original file line number Diff line number Diff line change
@@ -265,7 +265,7 @@ def main(cfg: TranscriptionConfig) -> Union[TranscriptionConfig, List[Hypothesis
asr_model.encoder.set_default_att_context_size(cfg.att_context_size)

# Setup decoding strategy
if hasattr(asr_model, 'change_decoding_strategy'):
if hasattr(asr_model, 'change_decoding_strategy') and hasattr(asr_model, 'decoding'):
if isinstance(asr_model.decoding, MultiTaskDecoding):
cfg.multitask_decoding.compute_langs = cfg.compute_langs
cfg.multitask_decoding.preserve_alignments = cfg.preserve_alignment

0 comments on commit 370477b

Please sign in to comment.