From de72dcf018ddc6650cb6a75ffdc2506c18c0b3d1 Mon Sep 17 00:00:00 2001
From: =?UTF-8?q?=E9=BB=84=E6=B5=B7?= <10402852@qq.com>
Date: Wed, 3 Jul 2024 16:08:33 +0800
Subject: [PATCH] 'commit'

---
 BaiHu/doTask.py | 12 +++---------
 1 file changed, 3 insertions(+), 9 deletions(-)

diff --git a/BaiHu/doTask.py b/BaiHu/doTask.py
index e80949d8..3fb6c3ac 100644
--- a/BaiHu/doTask.py
+++ b/BaiHu/doTask.py
@@ -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)