Timothy J. Baek 8 月之前
父节点
当前提交
85b4129219
共有 1 个文件被更改,包括 2 次插入1 次删除
  1. 2 1
      backend/apps/images/main.py

+ 2 - 1
backend/apps/images/main.py

@@ -298,7 +298,8 @@ def get_models(user=Depends(get_verified_user)):
 
 
             for node in app.state.config.COMFYUI_WORKFLOW_NODES:
             for node in app.state.config.COMFYUI_WORKFLOW_NODES:
                 if node["type"] == "model":
                 if node["type"] == "model":
-                    model_node_id = node["node_ids"][0]
+                    if node["node_ids"]:
+                        model_node_id = node["node_ids"][0]
                     break
                     break
 
 
             if model_node_id:
             if model_node_id: