| funasr/auto/auto_model.py | ●●●●● 补丁 | 查看 | 原始文档 | blame | 历史 |
funasr/auto/auto_model.py
@@ -398,7 +398,7 @@ if self.spk_model is not None: all_segments = sorted(all_segments, key=lambda x: x[0]) spk_embedding = result['spk_embedding'] labels = self.cb_model(spk_embedding, oracle_num=self.preset_spk_num) labels = self.cb_model(spk_embedding.cpu(), oracle_num=self.preset_spk_num) del result['spk_embedding'] sv_output = postprocess(all_segments, None, labels, spk_embedding.cpu()) if self.spk_mode == 'vad_segment':
