Browse Source

chore: format

Timothy J. Baek 7 tháng trước cách đây
mục cha
commit
7df997c992
2 tập tin đã thay đổi với 19 bổ sung9 xóa
  1. 17 6
      backend/open_webui/apps/images/main.py
  2. 2 3
      backend/open_webui/config.py

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

@@ -144,10 +144,22 @@ async def update_config(form_data: ConfigForm, user=Depends(get_admin_user)):
     app.state.config.AUTOMATIC1111_API_AUTH = (
         form_data.automatic1111.AUTOMATIC1111_API_AUTH
     )
-    
-    app.state.config.AUTOMATIC1111_CFG_SCALE = float(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 != "" else None
-    app.state.config.AUTOMATIC1111_SCHEDULER = form_data.automatic1111.AUTOMATIC1111_SCHEDULER if form_data.automatic1111.AUTOMATIC1111_SCHEDULER != "" else None
+
+    app.state.config.AUTOMATIC1111_CFG_SCALE = (
+        float(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 != ""
+        else None
+    )
+    app.state.config.AUTOMATIC1111_SCHEDULER = (
+        form_data.automatic1111.AUTOMATIC1111_SCHEDULER
+        if form_data.automatic1111.AUTOMATIC1111_SCHEDULER != ""
+        else None
+    )
 
     app.state.config.COMFYUI_BASE_URL = form_data.comfyui.COMFYUI_BASE_URL.strip("/")
     app.state.config.COMFYUI_WORKFLOW = form_data.comfyui.COMFYUI_WORKFLOW
@@ -285,7 +297,6 @@ async def update_image_config(form_data: ImageConfigForm, user=Depends(get_admin
             detail=ERROR_MESSAGES.INCORRECT_FORMAT("  (e.g., 50)."),
         )
 
-
     return {
         "MODEL": app.state.config.MODEL,
         "IMAGE_SIZE": app.state.config.IMAGE_SIZE,
@@ -543,7 +554,7 @@ async def image_generations(
 
             if form_data.negative_prompt is not None:
                 data["negative_prompt"] = form_data.negative_prompt
-            
+
             if app.state.config.AUTOMATIC1111_CFG_SCALE:
                 data["cfg_scale"] = app.state.config.AUTOMATIC1111_CFG_SCALE
 

+ 2 - 3
backend/open_webui/config.py

@@ -1236,8 +1236,7 @@ AUTOMATIC1111_SAMPLER = PersistentConfig(
         os.environ.get("AUTOMATIC1111_SAMPLER")
         if os.environ.get("AUTOMATIC1111_SAMPLER")
         else None
-    )
-   
+    ),
 )
 
 AUTOMATIC1111_SCHEDULER = PersistentConfig(
@@ -1247,7 +1246,7 @@ AUTOMATIC1111_SCHEDULER = PersistentConfig(
         os.environ.get("AUTOMATIC1111_SCHEDULER")
         if os.environ.get("AUTOMATIC1111_SCHEDULER")
         else None
-    )
+    ),
 )
 
 COMFYUI_BASE_URL = PersistentConfig(