zhifu gao
2023-01-18 2de29621c5b75fac8324ab9e04f0ce59c09a86b6
funasr/bin/asr_inference_uniasr.py
@@ -96,11 +96,14 @@
        else:
            decoder = asr_model.decoder2
        ctc = CTCPrefixScorer(ctc=asr_model.ctc, eos=asr_model.eos)
        if asr_model.ctc != None:
            ctc = CTCPrefixScorer(ctc=asr_model.ctc, eos=asr_model.eos)
            scorers.update(
                ctc=ctc
            )
        token_list = asr_model.token_list
        scorers.update(
            decoder=decoder,
            ctc=ctc,
            length_bonus=LengthBonus(len(token_list)),
        )