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

ADD midjourney修图接口支持多图返回

This commit is contained in:
kyj@bowong.ai 2025-07-16 19:03:04 +08:00
parent bbea095da9
commit 13f0176465
1 changed files with 8 additions and 1 deletions
Show Stats Download Patch File Download Diff File Expand all files Collapse all files

7
nodes/image_modal_nodes.py
View File

@ -239,6 +239,13 @@ class ModalMidJourneyGenerateImage:
if "fail" in resp.json()["msg"]: if "fail" in resp.json()["msg"]:
raise Exception("生成失败，可能因为风控") raise Exception("生成失败，可能因为风控")
result_url = resp.json()["data"] result_url = resp.json()["data"]
if isinstance(result_url, list):
result_list = []
for url in result_url:
logger.success("img_url: "+url)
result_list.append(url_to_tensor(url).squeeze(0))
result_list = torch.stack(result_list, dim=0)
return (result_list,)
logger.success("img_url: " + result_url) logger.success("img_url: " + result_url)
return (url_to_tensor(result_url),) return (url_to_tensor(result_url),)
sleep(interval) sleep(interval)