diff --git a/BaiHu/doTask.py b/BaiHu/doTask.py
index 0ac0285a..7d5bb12d 100644
--- a/BaiHu/doTask.py
+++ b/BaiHu/doTask.py
@@ -215,7 +215,8 @@ def runComfyUI(model_id, json_data, input_image, target_folder):
         #  # 生成下载链接
         url = 'http://hzkc.oss-cn-beijing.aliyuncs.com/' + key
         res.append(url)
-
+    # 清理GPU缓存
+    clear_cache(server_address)
 
 # 获取任务
 def get_task():
@@ -289,8 +290,7 @@ def get_task():
             if model_type_id == 3:
                 runComfyUI(model_id=model_id, json_data=vjson_data, input_image=input_image,
                            target_folder=output_path)
-                # 清理GPU缓存
-                clear_cache(server_address)
+
 
             return task_id
         elif success == 0: