diff --git a/xinference/model/embedding/core.py b/xinference/model/embedding/core.py index 8bb6c05626..67e4fa4bb4 100644 --- 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 )