shixian.shi
2024-02-06 410a85402db06bf36a9d7acec5dc922012951242
bug fix for numpy input
1个文件已修改
2 ■■■ 已修改文件
funasr/models/campplus/model.py 2 ●●● 补丁 | 查看 | 原始文档 | blame | 历史
funasr/models/campplus/model.py
@@ -123,5 +123,5 @@
        time3 = time.perf_counter()
        meta_data["extract_feat"] = f"{time3 - time2:0.3f}"
        meta_data["batch_data_time"] = np.array(speech_times).sum().item() / 16000.0
        results = [{"spk_embedding": self.forward(speech)}]
        results = [{"spk_embedding": self.forward(speech.to(torch.float32))}]
        return results, meta_data