UniqAI/open-webui
1
0
Fork 0
mirror of https://github.com/open-webui/open-webui synced 2024-11-16 05:24:02 +00:00
Code Issues Actions Packages Projects Releases Wiki Activity

refac: openai image async gen

Browse Source
This commit is contained in:
Timothy J. Baek 2024-09-20 03:16:08 +02:00
parent bd09b6dbad
commit d12b6cda4e
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/open_webui/apps/images/main.py
View File

@ -470,7 +470,9 @@ async def image_generations(
"response_format": "b64_json", "response_format": "b64_json",
} }
r = requests.post( # Use asyncio.to_thread for the requests.post call
r = await asyncio.to_thread(
requests.post,
url=f"{app.state.config.OPENAI_API_BASE_URL}/images/generations", url=f"{app.state.config.OPENAI_API_BASE_URL}/images/generations",
json=data, json=data,
headers=headers, headers=headers,