Procházet zdrojové kódy

Update asr_infer.py (#622)

bug fix
aky15 před 2 roky
rodič
revize
68ae83684f
1 změnil soubory, kde provedl 2 přidání a 2 odebrání
  1. 2 2
      funasr/bin/asr_infer.py

+ 2 - 2
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]