From f291890d9629a174ad8e1d3f1f0087e74f709842 Mon Sep 17 00:00:00 2001 From: qinxuye Date: Wed, 23 Oct 2024 21:12:24 +0800 Subject: [PATCH] fix --- xinference/model/embedding/core.py | 7 ++++++- 1 file changed, 6 insertions(+), 1 deletion(-) 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 )