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

View File

@ -239,6 +239,13 @@ class ModalMidJourneyGenerateImage:
if "fail" in resp.json()["msg"]:
raise Exception("生成失败,可能因为风控")
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)
return (url_to_tensor(result_url),)
sleep(interval)