shixian.shi
2023-03-09 3610b3f48ad4fca2178686b011a3ed0ea2e52c6a
update frontend in pipeline
1个文件已修改
2 ■■■ 已修改文件
funasr/bin/tp_inference.py 2 ●●● 补丁 | 查看 | 原始文档 | blame | 历史
funasr/bin/tp_inference.py
@@ -148,11 +148,11 @@
        # 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