'commit'

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...")