王梦迪
2025-05-20 fe588bc508c0076bb007d6ed36c18ac8ecb341ac
funasr/models/language_model/rnn/decoders.py
@@ -873,8 +873,10 @@
                        ctc_state[idx], accum_best_ids
                    )
        with torch.cuda.device(vscores.device):
            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 = [