'commit'

BaiHu/Config.ini

@@ -15,6 +15,7 @@ webui_server = http://192.168.1.21:7860
 txt2img_url = /sdapi/v1/txt2img
 img2img_url = /sdapi/v1/img2img
 get_vram_url = /sdapi/v1/get_vram
+empty_cache_url = /sdapi/v1/empty_cache
 ; WEB服务器地址
 [webServer]
 ; web_url = https://www.hzkjai.com

BaiHu/Tools/TestComfyUIApi.py

@@ -5,12 +5,9 @@ from Util.ComfyUIUtil import *
 config = ConfigUtil.getConfig()
 server_address = config.get('comfyui', 'server_address')
 
-# 清理GPU缓存
-# info = getUse(server_address)
-# print(info)
-# clear_cache(server_address)
 
-def get_vram():
+# 获取webui使用情况
+def get_webui_used():
     # webui 服务器地址
     webui_server = config['webui']['webui_server']
     vram_url = webui_server + config['webui']['get_vram_url']
@@ -20,5 +17,29 @@ def get_vram():
     else:
         print("调用集成的获取显卡显存代码失败！")
 
-get_vram()
 
+# 清理webui的缓存
+def clear_webui_cache():
+    # webui 服务器地址
+    webui_server = config['webui']['webui_server']
+    empty_cache_url = webui_server + config['webui']['empty_cache_url']
+    res = submit_post(empty_cache_url, None)
+    if res.status_code == 200:
+        print(res.text)
+    else:
+        print("调用集成的清理显卡显存代码失败！")
+
+
+# 清理GPU缓存
+info = get_comfyui_used(server_address)
+print(info)
+clear_comfyui_cache(server_address)
+info = get_comfyui_used(server_address)
+print(info)
+
+# 获取使显存数量
+get_webui_used()
+# 清理缓存
+clear_webui_cache()
+# 获取使显存数量
+get_webui_used()

BaiHu/Util/ComfyUIUtil.py

@@ -85,7 +85,7 @@ def get_images(ws, server_address, client_id, prompt):
 
 
 # 获取显卡使用率
-def getUse(server_address):
+def get_comfyui_used(server_address):
     req = urllib.request.Request("http://{}/system_stats".format(server_address))
     res = json.loads(urllib.request.urlopen(req).read())
 
@@ -101,9 +101,9 @@ def getUse(server_address):
 # 参考文档：如何清空Comfyui的gpu缓存
 # https://wailikeji.blog.csdn.net/article/details/140035515
 # 清理GPU显存
-def clear_cache(server_address):
+def clear_comfyui_cache(server_address):
     # 显示显卡使用率
-    print('清理显存前：' + getUse(server_address))
+    print('清理显存前：' + get_comfyui_used(server_address))
     # 打开文件并读取内容
     file_path = r'../JSON/clearGPU.json'
     if not os.path.exists(file_path):
@@ -115,7 +115,7 @@ def clear_cache(server_address):
 
     # 显示显卡使用率
     time.sleep(2)
-    print('清理显存后：' + getUse(server_address))
+    print('清理显存后：' + get_comfyui_used(server_address))
 
 
 # 生成图像
@@ -140,7 +140,7 @@ def generate_clip(server_address, prompt_data, client_id, output_path, myfilter)
                 binary_file.write(image_data)
     ws.close()
     # 清理GPU缓存
-    clear_cache(server_address)
+    clear_comfyui_cache(server_address)
     return files