diff --git a/apps/setting/models_provider/impl/xinference_model_provider/xinference_model_provider.py b/apps/setting/models_provider/impl/xinference_model_provider/xinference_model_provider.py index 0e432e061..ae2dc776c 100644 --- a/apps/setting/models_provider/impl/xinference_model_provider/xinference_model_provider.py +++ b/apps/setting/models_provider/impl/xinference_model_provider/xinference_model_provider.py @@ -407,4 +407,4 @@ class XinferenceModelProvider(IModelProvider): def get_model_info_by_name(model_list, model_name): if model_list is None: return [] - return [model for model in model_list if model.get('model_name') == model_name] + return [model for model in model_list if model.get('model_name') == model_name or model.get('id') == model_name]