Reset VAE upon restart

This commit is contained in:
cmdr2 2024-10-01 16:25:24 +05:30
parent 754a5f5e52
commit f51ab909ff

View File

@ -13,7 +13,12 @@ from sdkit.utils import base64_str_to_img, img_to_base64_str
WEBUI_HOST = "localhost" WEBUI_HOST = "localhost"
WEBUI_PORT = "7860" WEBUI_PORT = "7860"
DEFAULT_WEBUI_OPTIONS = {"show_progress_every_n_steps": 3, "show_progress_grid": True, "live_previews_enable": False} DEFAULT_WEBUI_OPTIONS = {
"show_progress_every_n_steps": 3,
"show_progress_grid": True,
"live_previews_enable": False,
"forge_additional_modules": [],
}
webui_opts: dict = None webui_opts: dict = None
@ -449,6 +454,8 @@ def image_progress_thread(task_id, callback, stream_image_progress, total_images
) )
if res.status_code == 200: if res.status_code == 200:
res = res.json() res = res.json()
else:
raise RuntimeError(f"Unexpected progress response. Status code: {res.status_code}. Res: {res.text}")
last_preview_id = res["id_live_preview"] last_preview_id = res["id_live_preview"]