Debug(infer_tool.py) debug feature index

This commit is contained in:
YuriHead 2023-06-04 00:37:38 +08:00 committed by GitHub
parent 0067f8120b
commit 4f9e440f04
No known key found for this signature in database
GPG Key ID: 4AEE18F83AFDEB23
1 changed files with 2 additions and 2 deletions

View File

@ -232,7 +232,7 @@ class Svc(object):
speaker_id = speaker
feature_index = self.cluster_model[speaker_id]
feat_np = c.transpose(0,1).cpu().numpy()
if self.big_npy is not None or self.now_spk_id != speaker_id:
if self.big_npy is None or self.now_spk_id != speaker_id:
self.big_npy = feature_index.reconstruct_n(0, feature_index.ntotal)
self.now_spk_id = speaker_id
print("starting feature retrieval...")
@ -530,4 +530,4 @@ class RealTimeVC:
self.last_chunk = audio[-self.pre_len:]
self.last_o = audio
return ret[self.chunk_len:2 * self.chunk_len]