Переглянути джерело

refac: gpt-image-1 support

Timothy Jaeryang Baek 5 місяців тому
батько
коміт
6ae2d2c4ea
1 змінених файлів з 5 додано та 1 видалено
  1. 5 1
      backend/open_webui/routers/images.py

+ 5 - 1
backend/open_webui/routers/images.py

@@ -500,7 +500,11 @@ async def image_generations(
                     if form_data.size
                     else request.app.state.config.IMAGE_SIZE
                 ),
-                "response_format": "b64_json",
+                **(
+                    {"response_format": "b64_json"}
+                    if "gpt-image-1" in request.app.state.config.IMAGE_GENERATION_MODEL
+                    else {}
+                ),
             }
 
             # Use asyncio.to_thread for the requests.post call