OpenWebUI/open-webui
1
0
Fork 0
mirror of https://github.com/open-webui/open-webui synced 2025-05-17 03:54:02 +00:00
Code Issues Actions 4 Packages Projects Releases Wiki Activity

fix: automatic1111 model update issue

Browse Source
This commit is contained in:
Timothy J. Baek 2024-09-04 15:25:31 +02:00
parent 9bcbf5e9b3
commit 76806a998f
1 changed files with 3 additions and 1 deletions
Show Stats Download Patch File Download Diff File Expand all files Collapse all files

4
backend/apps/images/main.py
View File

@ -192,6 +192,7 @@ async def verify_url(user=Depends(get_admin_user)):
def set_image_model(model: str): def set_image_model(model: str):
log.info(f"Setting image model to {model}")
app.state.config.MODEL = model app.state.config.MODEL = model
if app.state.config.ENGINE in ["", "automatic1111"]: if app.state.config.ENGINE in ["", "automatic1111"]:
api_auth = get_automatic1111_api_auth() 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") @app.post("/image/config/update")
async def update_image_config(form_data: ImageConfigForm, user=Depends(get_admin_user)): 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+$" pattern = r"^\d+x\d+$"
if re.match(pattern, form_data.IMAGE_SIZE): if re.match(pattern, form_data.IMAGE_SIZE):