Update index.py

2025-05-18 18:17:13 +02:00 · 2025-05-18 18:17:13 +02:00 · 96a083fd72
commit 96a083fd72
parent 58f7cbf669
1 changed files with 1 additions and 1 deletions
--- a/index.py
+++ b/index.py
@ -47,7 +47,7 @@ def main():
    print(f"Découpé {len(docs)} passages, génération des embeddings…")

    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.add(emb.astype("float32"))