Spaces:
Runtime error
Runtime error
| # src/utils/faiss_utils.py | |
| import faiss | |
| import numpy as np | |
| def build_faiss_index(embeddings: np.ndarray, use_gpu=False): | |
| d = embeddings.shape[1] | |
| index = faiss.IndexFlatIP(d) # Inner Product (cosine similarity if normalized) | |
| if use_gpu: | |
| res = faiss.StandardGpuResources() | |
| index = faiss.index_cpu_to_gpu(res, 0, index) | |
| index.add(embeddings.astype(np.float32)) | |
| return index | |
| def save_faiss_index(index, path): | |
| faiss.write_index(index, path) |