Browse Source

Merge pull request #14165 from Zyfax/patch-1

fix: image model list
Tim Jaeryang Baek 1 month ago
parent
commit
336d0b45ad
1 changed files with 2 additions and 1 deletions
  1. 2 1
      backend/open_webui/routers/images.py

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

@@ -333,10 +333,11 @@ def get_models(request: Request, user=Depends(get_verified_user)):
             return [
                 {"id": "dall-e-2", "name": "DALL·E 2"},
                 {"id": "dall-e-3", "name": "DALL·E 3"},
+                {"id": "gpt-image-1", "name": "GPT-IMAGE 1"},
             ]
         elif request.app.state.config.IMAGE_GENERATION_ENGINE == "gemini":
             return [
-                {"id": "imagen-3-0-generate-002", "name": "imagen-3.0 generate-002"},
+                {"id": "imagen-3.0-generate-002", "name": "imagen-3.0 generate-002"},
             ]
         elif request.app.state.config.IMAGE_GENERATION_ENGINE == "comfyui":
             # TODO - get models from comfyui