瀏覽代碼

Merge pull request #5537 from open-webui/dev

refac
Timothy Jaeryang Baek 7 月之前
父節點
當前提交
d4df552076
共有 1 個文件被更改,包括 6 次插入4 次删除
  1. 6 4
      backend/open_webui/apps/images/main.py

+ 6 - 4
backend/open_webui/apps/images/main.py

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