From 840a2cd7334c81359e2ceaed7a8a6eeedd2f409c Mon Sep 17 00:00:00 2001
From: wxywb <xy.wang@zilliz.com>
Date: Wed, 27 Nov 2024 20:04:07 +0800
Subject: [PATCH] fix: Change BGEM3 parameter to 'devices'. (#51)

Signed-off-by: wxywb <xy.wang@zilliz.com>
---
 milvus_model/hybrid/bge_m3.py | 2 +-
 1 file changed, 1 insertion(+), 1 deletion(-)

diff --git a/milvus_model/hybrid/bge_m3.py b/milvus_model/hybrid/bge_m3.py
index ef2dc4f..22f8d24 100644
--- a/milvus_model/hybrid/bge_m3.py
+++ b/milvus_model/hybrid/bge_m3.py
@@ -55,7 +55,7 @@ def __init__(
         _model_config = dict(
             {
                 "model_name_or_path": model_name,
-                "device": device,
+                "devices": device,
                 "normalize_embeddings": normalize_embeddings,
                 "use_fp16": use_fp16,
             },