From 3c74ce3653a5584713759f8758acbaf70bf0dd45 Mon Sep 17 00:00:00 2001 From: Timothy Jaeryang Baek Date: Wed, 25 Dec 2024 23:34:54 -0800 Subject: [PATCH] fix: automatic1111 image generation --- backend/open_webui/routers/images.py | 6 +++--- 1 file changed, 3 insertions(+), 3 deletions(-) diff --git a/backend/open_webui/routers/images.py b/backend/open_webui/routers/images.py index a42a7be5b..c40d12522 100644 --- a/backend/open_webui/routers/images.py +++ b/backend/open_webui/routers/images.py @@ -236,7 +236,7 @@ def get_image_model(request): try: r = requests.get( url=f"{request.app.state.config.AUTOMATIC1111_BASE_URL}/sdapi/v1/options", - headers={"authorization": get_automatic1111_api_auth()}, + headers={"authorization": get_automatic1111_api_auth(request)}, ) options = r.json() return options["sd_model_checkpoint"] @@ -354,7 +354,7 @@ def get_models(request: Request, user=Depends(get_verified_user)): ): r = requests.get( url=f"{request.app.state.config.AUTOMATIC1111_BASE_URL}/sdapi/v1/sd-models", - headers={"authorization": get_automatic1111_api_auth()}, + headers={"authorization": get_automatic1111_api_auth(request)}, ) models = r.json() return list( @@ -578,7 +578,7 @@ async def image_generations( requests.post, url=f"{request.app.state.config.AUTOMATIC1111_BASE_URL}/sdapi/v1/txt2img", json=data, - headers={"authorization": get_automatic1111_api_auth()}, + headers={"authorization": get_automatic1111_api_auth(request)}, ) res = r.json()