fix: Set default device in BGEM3EmbeddingFunction to None. (#48)

milvus-io · Nov 20, 2024 · 24db758 · 24db758
1 parent 1b9549e
commit 24db758
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
@@ -29,7 +29,7 @@ def __init__(
         self,
         model_name: str = "BAAI/bge-m3",
         batch_size: int = 16,
-        device: str = "",
+        device: str = None,
         normalize_embeddings: bool = True,
         use_fp16: bool = False,
         return_dense: bool = True,