UniqAI/open-webui
1
0
Fork 0
mirror of https://github.com/open-webui/open-webui synced 2025-01-18 08:40:58 +00:00
Code Issues Actions 5 Packages Projects Releases Wiki Activity

Merge pull request #5537 from open-webui/dev

Browse Source 
refac
This commit is contained in:
Timothy Jaeryang Baek 2024-09-20 03:18:40 +02:00 committed by GitHub
commit d4df552076
No known key found for this signature in database
GPG Key ID: B5690EEEBB952194
1 changed files with 6 additions and 4 deletions
Show Stats Download Patch File Download Diff File Expand all files Collapse all files

10
backend/open_webui/apps/images/main.py
View File

@ -147,17 +147,17 @@ async def update_config(form_data: ConfigForm, user=Depends(get_admin_user)):
app.state.config.AUTOMATIC1111_CFG_SCALE = ( app.state.config.AUTOMATIC1111_CFG_SCALE = (
float(form_data.automatic1111.AUTOMATIC1111_CFG_SCALE) float(form_data.automatic1111.AUTOMATIC1111_CFG_SCALE)
if form_data.automatic1111.AUTOMATIC1111_CFG_SCALE != "" if form_data.automatic1111.AUTOMATIC1111_CFG_SCALE
else None else None
) )
app.state.config.AUTOMATIC1111_SAMPLER = ( app.state.config.AUTOMATIC1111_SAMPLER = (
form_data.automatic1111.AUTOMATIC1111_SAMPLER form_data.automatic1111.AUTOMATIC1111_SAMPLER
if form_data.automatic1111.AUTOMATIC1111_SAMPLER != "" if form_data.automatic1111.AUTOMATIC1111_SAMPLER
else None else None
) )
app.state.config.AUTOMATIC1111_SCHEDULER = ( app.state.config.AUTOMATIC1111_SCHEDULER = (
form_data.automatic1111.AUTOMATIC1111_SCHEDULER form_data.automatic1111.AUTOMATIC1111_SCHEDULER
if form_data.automatic1111.AUTOMATIC1111_SCHEDULER != "" if form_data.automatic1111.AUTOMATIC1111_SCHEDULER
else None else None
) )
@ -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,