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("等待超时")