From 58ee017fca6d718034e2a387854e12f8c4d10b42 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:09:19 +0800
Subject: [PATCH] 'commit'

---
 BaiHu/Util/SDUtil.py | 4 +++-
 1 file changed, 3 insertions(+), 1 deletion(-)

diff --git a/BaiHu/Util/SDUtil.py b/BaiHu/Util/SDUtil.py
index 459f5e5f..8c0149d9 100644
--- a/BaiHu/Util/SDUtil.py
+++ b/BaiHu/Util/SDUtil.py
@@ -206,9 +206,11 @@ def restart_server(webui_address):
 
 # 清理一下SD
 def release_sd(webui_address,comfyui_address):
+    # comfyui的显存先清理一下
+    clear_comfyui_cache(comfyui_address)
+
     # 获取使显存数量
     total, used = get_webui_used(webui_address)
-    clear_comfyui_cache(comfyui_address)
     # 显存占用超过20GB
     if float(used) > 20:
         print("显存占用高达" + used + "GB，准备重启SD...")