fix: Change BGEM3 parameter to 'devices'. (#51)

Signed-off-by: wxywb <xy.wang@zilliz.com>
milvus-io · Nov 27, 2024 · 840a2cd · 840a2cd
1 parent 91e34fa
commit 840a2cd
Showing 1 changed file with 1 addition and 1 deletion.
diff --git 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,
             },