|
|
@ -198,6 +198,9 @@ def get_task():
|
|
|
|
if response.status_code == 200:
|
|
|
|
if response.status_code == 200:
|
|
|
|
success = response.json()["success"]
|
|
|
|
success = response.json()["success"]
|
|
|
|
if success == 1:
|
|
|
|
if success == 1:
|
|
|
|
|
|
|
|
# 清理服务器显存
|
|
|
|
|
|
|
|
release_sd(webui_address, comfyui_address)
|
|
|
|
|
|
|
|
|
|
|
|
data = response.json()["data"]
|
|
|
|
data = response.json()["data"]
|
|
|
|
# 任务编号
|
|
|
|
# 任务编号
|
|
|
|
task_id = data['task_id']
|
|
|
|
task_id = data['task_id']
|
|
|
@ -314,9 +317,6 @@ if __name__ == '__main__':
|
|
|
|
|
|
|
|
|
|
|
|
while True:
|
|
|
|
while True:
|
|
|
|
try:
|
|
|
|
try:
|
|
|
|
# 清理服务器显存
|
|
|
|
|
|
|
|
release_sd(webui_address, comfyui_address)
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
res = [] # 所有图片的地址路径
|
|
|
|
res = [] # 所有图片的地址路径
|
|
|
|
v_task_id = get_task()
|
|
|
|
v_task_id = get_task()
|
|
|
|
# 回复上位生成完毕
|
|
|
|
# 回复上位生成完毕
|
|
|
|