'commit'

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)
 

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，清理一下缓存...")