diff --git a/BaiHu/Tools/TestComfyUIApi.py b/BaiHu/Tools/TestComfyUIApi.py
index 68daba75..ae4c199c 100644
--- a/BaiHu/Tools/TestComfyUIApi.py
+++ b/BaiHu/Tools/TestComfyUIApi.py
@@ -6,13 +6,6 @@ config = ConfigUtil.getConfig()
 comfyui_address = config.get('comfyui', 'server_address')
 webui_address = config.get('webui', 'webui_address')
 
-# 清理GPU缓存
-info = get_comfyui_used(comfyui_address)
-print(info)
-clear_comfyui_cache(comfyui_address)
-info = get_comfyui_used(comfyui_address)
-print(info)
-
 # 清理一下sd
-release_sd()
+release_sd(webui_address,comfyui_address)
 
diff --git a/BaiHu/Util/SDUtil.py b/BaiHu/Util/SDUtil.py
index a73d5695..aae960e8 100644
--- a/BaiHu/Util/SDUtil.py
+++ b/BaiHu/Util/SDUtil.py
@@ -205,9 +205,10 @@ def restart_server(webui_address):
 
 
 # 清理一下SD
-def release_sd(webui_address):
+def release_sd(webui_address,comfyui_address):
     # 获取使显存数量
     total, used = get_webui_used(webui_address)
+    clear_comfyui_cache(comfyui_address)
     # 显存占用超过20GB
     if float(used) > 20:
         print("显存占用高达" + used + "GB，准备重启SD...")
@@ -218,4 +219,5 @@ def release_sd(webui_address):
     else:
         # 清理一下缓存即可
         clear_webui_cache(webui_address)
+        clear_comfyui_cache(comfyui_address)
         print("显存占用正常，无需重新启动SD，清理一下缓存...")
diff --git a/BaiHu/Util/__pycache__/SDUtil.cpython-310.pyc b/BaiHu/Util/__pycache__/SDUtil.cpython-310.pyc
index 48655d96..936b15bf 100644
Binary files a/BaiHu/Util/__pycache__/SDUtil.cpython-310.pyc and b/BaiHu/Util/__pycache__/SDUtil.cpython-310.pyc differ