You can not select more than 25 topics
Topics must start with a letter or number, can include dashes ('-') and can be up to 35 characters long.
This file contains ambiguous Unicode characters that may be confused with others in your current locale. If your use case is intentional and legitimate, you can safely ignore this warning. Use the Escape button to highlight these characters.
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 )
# 获取使显存数量
total , used = get_webui_used ( webui_address )
# 显卡显存占用超过20GB
if float ( used ) > 20 :
print ( " 发现显存占用高达 " + used + " GB， " )
# 重新启动一下SD
restart_server ( webui_address )
# 休息2分钟以保证SD重启完成
time . sleep ( 2 * 60 )
else :
# 清理一下缓存即可
clear_webui_cache ( webui_address )
print ( " 显存占用正常， ， " )
