Explorar o código

update frontend in pipeline

shixian.shi %!s(int64=3) %!d(string=hai) anos
pai
achega
3610b3f48a
Modificáronse 1 ficheiros con 1 adicións e 1 borrados
  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
         if isinstance(speech, np.ndarray):
             speech = torch.tensor(speech)
-
         if self.frontend is not None:
             feats, feats_len = self.frontend.forward(speech, speech_lengths)
             feats = to_device(feats, device=self.device)
             feats_len = feats_len.int()
+            self.tp_model.frontend = None
         else:
             feats = speech
             feats_len = speech_lengths