FabNum/Code
4
0
Fork 0
Code Issues 2 Pull Requests Actions Packages Projects 1 Releases Wiki Activity

Update index.py

Browse Source
This commit is contained in:
Stéphan Peccini 2025-05-18 18:17:13 +02:00
parent 58f7cbf669
commit 96a083fd72
1 changed files with 1 additions and 1 deletions
Show Stats Download Patch File Download Diff File Expand all files Collapse all files

2
index.py
View File

@ -47,7 +47,7 @@ def main():
print(f"Découpé {len(docs)} passages, génération des embeddings…") print(f"Découpé {len(docs)} passages, génération des embeddings…")
model = BGEM3FlagModel(MODEL_NAME, device="cpu") model = BGEM3FlagModel(MODEL_NAME, device="cpu")
emb = model.encode(docs, batch_size=64, normalize_embeddings=True) emb = model.encode(docs, batch_size=64, normalize=True)
index = faiss.IndexFlatIP(emb.shape[1]) index = faiss.IndexFlatIP(emb.shape[1])
index.add(emb.astype("float32")) index.add(emb.astype("float32"))