zhifu gao
2025-04-22 2c2fb5e1eb1185a081e3507c2aa5c3aafaa2bb6d
funasr/models/language_model/rnn/decoders.py
@@ -873,7 +873,10 @@
                        ctc_state[idx], accum_best_ids
                    )
        torch.cuda.empty_cache()
        device = vscores.device
        if device.type == 'cuda':
            with torch.cuda.device():
                torch.cuda.empty_cache()
        dummy_hyps = [{"yseq": [self.sos, self.eos], "score": np.array([-float("inf")])}]
        ended_hyps = [