from Util import ConfigUtil
from Util.ComfyUIUtil import *


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

# 显示显卡使用率
print('清理显存前：' + getUse(server_address))

# 清理GPU缓存
clearGPU(server_address)

# 显示显卡使用率
print('清理显存后：' + getUse(server_address))