From a0aa2f1d925a6d386717374357f4d782d616f52b 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:10:40 +0800
Subject: [PATCH] 'commit'

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

diff --git a/BaiHu/doTask.py b/BaiHu/doTask.py
index bacec740..3367103e 100644
--- a/BaiHu/doTask.py
+++ b/BaiHu/doTask.py
@@ -258,6 +258,9 @@ def get_task():
                 runComfyUI(model_id=model_id, json_data=vjson_data, input_image=input_image,
                            target_folder=output_path)
 
+            # 清理服务器显存
+            release_sd(webui_address, comfyui_address)
+
             return task_id
         elif success == 0:
             printf("现在没有待处理的任务,休息3秒后再试...")