KeCheng/BaiHu/Tools/TestComfyUIApi.py

from Util import ConfigUtil
from Util.SDUtil import *

# 打开配置文件
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()