From 68ae83684fea388fe6e51eebfd99414f55b6411d Mon Sep 17 00:00:00 2001 From: aky15 Date: Mon, 12 Jun 2023 21:41:41 +0800 Subject: [PATCH] Update asr_infer.py (#622) bug fix --- funasr/bin/asr_infer.py | 4 ++-- 1 file changed, 2 insertions(+), 2 deletions(-) diff --git a/funasr/bin/asr_infer.py b/funasr/bin/asr_infer.py index 0e203c4db..47ce0ee99 100644 --- a/funasr/bin/asr_infer.py +++ b/funasr/bin/asr_infer.py @@ -493,9 +493,9 @@ class Speech2TextParaformer: else: if pre_token_length[i] == 0: yseq = torch.tensor( - [self.asr_model.sos] + [self.asr_model.eos], device=yseq.device + [self.asr_model.sos] + [self.asr_model.eos], device=pre_acoustic_embeds.device ) - score = torch.tensor(0.0, device=yseq.device) + score = torch.tensor(0.0, device=pre_acoustic_embeds.device) else: yseq = am_scores.argmax(dim=-1) score = am_scores.max(dim=-1)[0]