ADD midjourney修图接口支持多图返回
This commit is contained in:
parent
bbea095da9
commit
13f0176465
|
|
@ -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)
|
||||
|
|
|
|||
Loading…
Reference in New Issue