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,
             },