From 22e51ec95fcba72e0a8016775b31ed9c60da3399 Mon Sep 17 00:00:00 2001 From: R1ckShi Date: Mon, 3 Jun 2024 15:56:39 +0800 Subject: [PATCH] bug fix --- runtime/python/libtorch/funasr_torch/paraformer_bin.py | 2 +- 1 file changed, 1 insertion(+), 1 deletion(-) diff --git a/runtime/python/libtorch/funasr_torch/paraformer_bin.py b/runtime/python/libtorch/funasr_torch/paraformer_bin.py index 755237e40..5fa3cc969 100644 --- a/runtime/python/libtorch/funasr_torch/paraformer_bin.py +++ b/runtime/python/libtorch/funasr_torch/paraformer_bin.py @@ -337,7 +337,7 @@ class ContextualParaformer(Paraformer): outputs = self.bb_infer(feats, feats_len, bias_embed) am_scores, valid_token_lens = outputs[0], outputs[1] else: - outputs = self.bb_infer_infer(feats.cuda(), feats_len.cuda(), bias_embed.cuda()) + outputs = self.bb_infer(feats.cuda(), feats_len.cuda(), bias_embed.cuda()) am_scores, valid_token_lens = outputs[0].cpu(), outputs[1].cpu() except: # logging.warning(traceback.format_exc())