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

https://github.com/1Panel-dev/MaxKB/issues/1389
2024-10-17 13:54:05 +08:00 · 2024-10-17 13:54:05 +08:00 · 8c4fb7968d
parent 735b7c7a22
commit 8c4fb7968d
1 changed files with 1 additions and 1 deletions
--- 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]