bowong
/
ComfyUI-CustomNode
6
0
Fork 0
Code Pull Requests Releases Activity

FIX

This commit is contained in:
kyj@bowong.ai 2025-07-25 16:43:46 +08:00
parent aa18570341
commit 2f38db1af5
2 changed files with 7 additions and 6 deletions
Show Stats Download Patch File Download Diff File Expand all files Collapse all files

6
__init__.py
View File

@ -85,9 +85,9 @@ NODE_DISPLAY_NAME_MAPPINGS = {
"Jinja2RenderTemplate": "Jinja2格式Prompt模板渲染", "Jinja2RenderTemplate": "Jinja2格式Prompt模板渲染",
"JMGestureCorrect": "人物侧身图片转为正面图-即梦", "JMGestureCorrect": "人物侧身图片转为正面图-即梦",
"ModalClothesMask": "模特指定衣服替换为指定颜色", "ModalClothesMask": "模特指定衣服替换为指定颜色",
"ModalEditCustom": "Prompt生/修图", "ModalEditCustom": "Prompt生/修图-Gemini",
"ModalMidJourneyGenerateImage": "Prompt生/修图", "ModalMidJourneyGenerateImage": "Prompt生/修图-MJ",
"ModalMidJourneyDescribeImage": "反推生图提示词", "ModalMidJourneyDescribeImage": "反推生图提示词-MJ",
"JMCustom": "Prompt生视频", "JMCustom": "Prompt生视频",
"VideoMerge":"顺序合并视频" "VideoMerge":"顺序合并视频"
} }

7
nodes/image_modal_nodes.py
View File

@ -1,5 +1,6 @@
import io import io
import json import json
import traceback
from time import sleep from time import sleep
import folder_paths import folder_paths
@ -241,8 +242,8 @@ class ModalMidJourneyGenerateImage:
timeout=150) timeout=150)
job_resp.raise_for_status() job_resp.raise_for_status()
job_resp = job_resp.json() job_resp = job_resp.json()
if "失败" in job_resp["msg"] or "fail" in job_resp["msg"] or "error" in job_resp["msg"]: if not job_resp["status"]:
raise Exception("生成失败") raise Exception("生成失败, 可能因为风控")
job_id = job_resp["data"] job_id = job_resp["data"]
for _ in range(0, timeout // interval, interval): for _ in range(0, timeout // interval, interval):
logger.info("查询结果") logger.info("查询结果")
@ -296,7 +297,7 @@ class ModalMidJourneyDescribeImage:
timeout=150) timeout=150)
job_resp.raise_for_status() job_resp.raise_for_status()
job_resp = job_resp.json() job_resp = job_resp.json()
if "失败" in job_resp["msg"] or "fail" in job_resp["msg"] or "error" in job_resp["msg"]: if not job_resp["status"]:
raise Exception("描述失败") raise Exception("描述失败")
result = job_resp["data"] result = job_resp["data"]
return (result,) return (result,)