decbc4d
1
2
3
4
5
6
7
8
9
10
11
12
import faiss import torch my_index = faiss.read_index("knn.index") query = torch.rand(3, 1024) k = 5 distances, indices = my_index.search(query, k) import pdb;pdb.set_trace()
