From 96a083fd7227388a3e272620fcc64ef94f33aa4a Mon Sep 17 00:00:00 2001 From: =?UTF-8?q?St=C3=A9phan?= Date: Sun, 18 May 2025 18:17:13 +0200 Subject: [PATCH] Update index.py --- index.py | 2 +- 1 file changed, 1 insertion(+), 1 deletion(-) diff --git a/index.py b/index.py index cfd34fe..ceab073 100644 --- 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"))