From 24db7589d2e1479902d280ca756396ecde08b4c8 Mon Sep 17 00:00:00 2001
From: wxywb <xy.wang@zilliz.com>
Date: Wed, 20 Nov 2024 19:40:14 +0800
Subject: [PATCH] fix: Set default device in BGEM3EmbeddingFunction to None.
 (#48)

---
 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 dc79ed7..d4dd34e 100644
--- 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,