main
黄海 1 year ago
parent 91ff27fb72
commit 458d01c15b

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

@ -140,8 +140,6 @@ def generate_clip(server_address, prompt_data, client_id, output_path, myfilter)
# 写入二进制文件 # 写入二进制文件
binary_file.write(image_data) binary_file.write(image_data)
ws.close() ws.close()
# 清理GPU缓存
clear_comfyui_cache(server_address)
return files return files
@ -169,10 +167,9 @@ def upload_file(server_address, file, subfolder="", overwrite=False):
# 获取webui使用情况 # 获取webui使用情况
def get_webui_used(config): def get_webui_used(webui_server):
# webui 服务器地址 # webui 服务器地址
webui_server = config['webui']['webui_server'] vram_url = webui_server + '/sdapi/v1/get_vram'
vram_url = webui_server + config['webui']['get_vram_url']
res = submit_post(vram_url, None) res = submit_post(vram_url, None)
if res.status_code == 200: if res.status_code == 200:
jo = json.loads(res.text) jo = json.loads(res.text)
@ -186,10 +183,9 @@ def get_webui_used(config):
# 清理webui的缓存 # 清理webui的缓存
def clear_webui_cache(config): def clear_webui_cache(webui_server):
# webui 服务器地址 # webui 服务器地址
webui_server = config['webui']['webui_server'] empty_cache_url = webui_server + '/sdapi/v1/empty_cache'
empty_cache_url = webui_server + config['webui']['empty_cache_url']
res = submit_post(empty_cache_url, None) res = submit_post(empty_cache_url, None)
if res.status_code == 200: if res.status_code == 200:
jo = json.loads(res.text) jo = json.loads(res.text)

@ -312,6 +312,14 @@ if __name__ == '__main__':
v_target_img_urls = v_target_img_urls[0:len(v_target_img_urls) - 1] v_target_img_urls = v_target_img_urls[0:len(v_target_img_urls) - 1]
# 生成完毕 # 生成完毕
finish_task(v_task_id, v_target_img_urls) finish_task(v_task_id, v_target_img_urls)
# 清理SD缓存
clear_webui_cache(server_address)
get_webui_used(server_address)
# 清理GPU缓存
clear_comfyui_cache(server_address)
get_comfyui_used(server_address)
elif v_task_id == 0: elif v_task_id == 0:
# 本轮没有任务 # 本轮没有任务
time.sleep(3) time.sleep(3)

Loading…
Cancel
Save