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"))