bgem3 devices kwargs syntax error bugfix (#50)

Closes #49 Co-authored-by: Sam Sweere <[email protected]>
milvus-io · Nov 27, 2024 · 91e34fa · 91e34fa
1 parent 24db758
commit 91e34fa
Showing 1 changed file with 3 additions and 3 deletions.
diff --git a/milvus_model/hybrid/bge_m3.py b/milvus_model/hybrid/bge_m3.py
@@ -49,13 +49,13 @@ def __init__(
             )
 
         if "devices" in kwargs:
-            device = devices
-            kwargs.pop(device)
+            device = kwargs["devices"]
+            kwargs.pop("devices")
 
         _model_config = dict(
             {
                 "model_name_or_path": model_name,
-                "devices": device,
+                "device": device,
                 "normalize_embeddings": normalize_embeddings,
                 "use_fp16": use_fp16,
             },