from Util import ConfigUtil
from Util.ComfyUIUtil import *

# 打开配置文件
config = ConfigUtil.getConfig()
server_address = config.get('comfyui', 'server_address')


# 清理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(config)
# 清理缓存
clear_webui_cache(config)
# 获取使显存数量
get_webui_used(config)