|
|
|
|
@ -314,6 +314,9 @@ if __name__ == '__main__':
|
|
|
|
|
|
|
|
|
|
while True:
|
|
|
|
|
try:
|
|
|
|
|
# 清理一下sd
|
|
|
|
|
release_sd(webui_address, comfyui_address)
|
|
|
|
|
|
|
|
|
|
res = [] # 所有图片的地址路径
|
|
|
|
|
v_task_id = get_task()
|
|
|
|
|
# 回复上位生成完毕
|
|
|
|
|
@ -325,15 +328,6 @@ if __name__ == '__main__':
|
|
|
|
|
v_target_img_urls = v_target_img_urls[0:len(v_target_img_urls) - 1]
|
|
|
|
|
# 生成完毕
|
|
|
|
|
finish_task(v_task_id, v_target_img_urls)
|
|
|
|
|
|
|
|
|
|
# 清理SD缓存
|
|
|
|
|
clear_webui_cache(webui_address)
|
|
|
|
|
total, used = get_webui_used(webui_address)
|
|
|
|
|
|
|
|
|
|
# 清理GPU缓存
|
|
|
|
|
clear_comfyui_cache(comfyui_address)
|
|
|
|
|
get_comfyui_used(comfyui_address)
|
|
|
|
|
|
|
|
|
|
elif v_task_id == 0:
|
|
|
|
|
# 本轮没有任务
|
|
|
|
|
time.sleep(3)
|
|
|
|
|
|
