From 13f0176465bcb471cc5c4536ff7ab6d73cb05e8e Mon Sep 17 00:00:00 2001 From: "kyj@bowong.ai" Date: Wed, 16 Jul 2025 19:03:04 +0800 Subject: [PATCH] =?UTF-8?q?ADD=20midjourney=E4=BF=AE=E5=9B=BE=E6=8E=A5?= =?UTF-8?q?=E5=8F=A3=E6=94=AF=E6=8C=81=E5=A4=9A=E5=9B=BE=E8=BF=94=E5=9B=9E?= MIME-Version: 1.0 Content-Type: text/plain; charset=UTF-8 Content-Transfer-Encoding: 8bit --- nodes/image_modal_nodes.py | 9 ++++++++- 1 file changed, 8 insertions(+), 1 deletion(-) diff --git a/nodes/image_modal_nodes.py b/nodes/image_modal_nodes.py index fefc3df..0d85528 100644 --- a/nodes/image_modal_nodes.py +++ b/nodes/image_modal_nodes.py @@ -239,7 +239,14 @@ class ModalMidJourneyGenerateImage: if "fail" in resp.json()["msg"]: raise Exception("生成失败，可能因为风控") result_url = resp.json()["data"] - logger.success("img_url: "+result_url) + 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) return (url_to_tensor(result_url),) sleep(interval) raise Exception("等待超时")