Skip to content

Commit 2fe97ac

Browse files
authored
Revert "fix model config "device" for BGEM3FlagModel" (#75)
Reverts #73 The latest BGEM3FlagModel receive `devices`: https://github.com/FlagOpen/FlagEmbedding/blob/master/FlagEmbedding/inference/embedder/encoder_only/m3.py
1 parent db974b3 commit 2fe97ac

1 file changed

Lines changed: 1 addition & 1 deletion

File tree

src/pymilvus/model/hybrid/bge_m3.py

Lines changed: 1 addition & 1 deletion
Original file line numberDiff line numberDiff line change
@@ -58,7 +58,7 @@ def __init__(
5858
_model_config = dict(
5959
{
6060
"model_name_or_path": model_name,
61-
"device": device,
61+
"devices": device,
6262
"normalize_embeddings": normalize_embeddings,
6363
"use_fp16": use_fp16,
6464
},

0 commit comments

Comments
 (0)