Skip to content

Commit dc0c875

Browse files
authored
Total_fea not needed now. Better and faster retrieval performance.
Total_fea not needed now. Better and faster retrieval performance.
1 parent 9be8048 commit dc0c875

File tree

1 file changed

+1
-0
lines changed

1 file changed

+1
-0
lines changed

vc_infer_pipeline.py

Lines changed: 1 addition & 0 deletions
Original file line numberDiff line numberDiff line change
@@ -123,6 +123,7 @@ def vc(
123123
# _, I = index.search(npy, 1)
124124
# npy = big_npy[I.squeeze()]
125125

126+
#by github @nadare881
126127
score, ix = index.search(npy, k=8)
127128
weight = np.square(1 / score)
128129
weight /= weight.sum(axis=1, keepdims=True)

0 commit comments

Comments
 (0)