From 76806a998f56d93e09a1bcf38c6ec4be15ca108c Mon Sep 17 00:00:00 2001 From: "Timothy J. Baek" Date: Wed, 4 Sep 2024 15:25:31 +0200 Subject: [PATCH] fix: automatic1111 model update issue --- backend/apps/images/main.py | 4 +++- 1 file changed, 3 insertions(+), 1 deletion(-) diff --git a/backend/apps/images/main.py b/backend/apps/images/main.py index 7af68f066..56c9a9a3a 100644 --- a/backend/apps/images/main.py +++ b/backend/apps/images/main.py @@ -192,6 +192,7 @@ async def verify_url(user=Depends(get_admin_user)): def set_image_model(model: str): + log.info(f"Setting image model to {model}") app.state.config.MODEL = model if app.state.config.ENGINE in ["", "automatic1111"]: api_auth = get_automatic1111_api_auth() @@ -245,7 +246,8 @@ async def get_image_config(user=Depends(get_admin_user)): @app.post("/image/config/update") async def update_image_config(form_data: ImageConfigForm, user=Depends(get_admin_user)): - app.state.config.MODEL = form_data.MODEL + + set_image_model(form_data.MODEL) pattern = r"^\d+x\d+$" if re.match(pattern, form_data.IMAGE_SIZE):