fix: 修复xinference模型创建失败的缺陷

https://github.com/1Panel-dev/MaxKB/issues/1389
This commit is contained in:
wxg0103 2024-10-17 13:54:05 +08:00 committed by wxg0103
parent 735b7c7a22
commit 8c4fb7968d

View File

@ -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]