fix

qinxuye · Oct 23, 2024 · f291890 · f291890
1 parent 89e6409
commit f291890
Showing 1 changed file with 6 additions and 1 deletion.
diff --git a/xinference/model/embedding/core.py b/xinference/model/embedding/core.py
@@ -477,7 +477,12 @@ def create_embedding_model_instance(
     if model_path is None:
         model_path = cache(model_spec)
 
-    model = EmbeddingModel(model_uid, model_path, model_spec, **kwargs)
+    from .mindie import MindIEEmbeddingModel, is_available
+
+    if is_available(model_spec):
+        model = MindIEEmbeddingModel(model_uid, model_path, model_spec, **kwargs)  # type: ignore
+    else:
+        model = EmbeddingModel(model_uid, model_path, model_spec, **kwargs)  # type: ignore
     model_description = EmbeddingModelDescription(
         subpool_addr, devices, model_spec, model_path=model_path
     )