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

Merge pull request #8095 from open-webui/dev

Browse Source 
fix: automatic1111 image generation
This commit is contained in:
Timothy Jaeryang Baek 2024-12-25 23:35:10 -08:00 committed by GitHub
commit dd933e406f
No known key found for this signature in database
GPG Key ID: B5690EEEBB952194
1 changed files with 3 additions and 3 deletions
Show Stats Download Patch File Download Diff File Expand all files Collapse all files

6
backend/open_webui/routers/images.py
View File

@ -236,7 +236,7 @@ def get_image_model(request):
try: try:
r = requests.get( r = requests.get(
url=f"{request.app.state.config.AUTOMATIC1111_BASE_URL}/sdapi/v1/options", 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() options = r.json()
return options["sd_model_checkpoint"] return options["sd_model_checkpoint"]
@ -354,7 +354,7 @@ def get_models(request: Request, user=Depends(get_verified_user)):
): ):
r = requests.get( r = requests.get(
url=f"{request.app.state.config.AUTOMATIC1111_BASE_URL}/sdapi/v1/sd-models", 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() models = r.json()
return list( return list(
@ -578,7 +578,7 @@ async def image_generations(
requests.post, requests.post,
url=f"{request.app.state.config.AUTOMATIC1111_BASE_URL}/sdapi/v1/txt2img", url=f"{request.app.state.config.AUTOMATIC1111_BASE_URL}/sdapi/v1/txt2img",
json=data, json=data,
headers={"authorization": get_automatic1111_api_auth()}, headers={"authorization": get_automatic1111_api_auth(request)},
) )
res = r.json() res = r.json()