zhuchaowe/maxkb
1
0
Fork 0
Code Issues Pull Requests Actions Packages Projects Releases Wiki Activity

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

Browse Source 
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
1 changed files with 1 additions and 1 deletions
Show Stats Download Patch File Download Diff File Expand all files Collapse all files

2
apps/setting/models_provider/impl/xinference_model_provider/xinference_model_provider.py
View File

@ -407,4 +407,4 @@ class XinferenceModelProvider(IModelProvider):
def get_model_info_by_name(model_list, model_name): def get_model_info_by_name(model_list, model_name):
if model_list is None: if model_list is None:
return [] 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]