shixian.shi 3 лет назад
Родитель
Сommit
3610b3f48a
1 измененных файлов с 1 добавлено и 1 удалено
  1. 1 1
      funasr/bin/tp_inference.py

+ 1 - 1
funasr/bin/tp_inference.py

@@ -148,11 +148,11 @@ class SpeechText2Timestamp:
         # Input as audio signal
         # Input as audio signal
         if isinstance(speech, np.ndarray):
         if isinstance(speech, np.ndarray):
             speech = torch.tensor(speech)
             speech = torch.tensor(speech)
-
         if self.frontend is not None:
         if self.frontend is not None:
             feats, feats_len = self.frontend.forward(speech, speech_lengths)
             feats, feats_len = self.frontend.forward(speech, speech_lengths)
             feats = to_device(feats, device=self.device)
             feats = to_device(feats, device=self.device)
             feats_len = feats_len.int()
             feats_len = feats_len.int()
+            self.tp_model.frontend = None
         else:
         else:
             feats = speech
             feats = speech
             feats_len = speech_lengths
             feats_len = speech_lengths