From 6bbf81f3f1cb607400dfed675357bf10632b9561 Mon Sep 17 00:00:00 2001
From: =?UTF-8?q?=E9=BB=84=E6=B5=B7?= <10402852@qq.com>
Date: Mon, 1 Jul 2024 09:09:18 +0800
Subject: [PATCH] 'commit'

---
 BaiHu/doTask.py | 2 ++
 1 file changed, 2 insertions(+)

diff --git a/BaiHu/doTask.py b/BaiHu/doTask.py
index a8109123..0ac0285a 100644
--- a/BaiHu/doTask.py
+++ b/BaiHu/doTask.py
@@ -289,6 +289,8 @@ 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: