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 = [