AI/open-webui
1
0
Fork 1
mirror of https://git.mirrors.martin98.com/https://github.com/open-webui/open-webui synced 2025-08-15 19:55:52 +08:00
Code Issues Packages Projects Releases Wiki Activity

Merge pull request #14165 from Zyfax/patch-1

Browse Source 
fix: image model list
This commit is contained in:
Tim Jaeryang Baek 2025-05-22 15:37:00 +04:00 committed by GitHub
commit 336d0b45ad
No known key found for this signature in database
GPG Key ID: B5690EEEBB952194
1 changed files with 2 additions and 1 deletions
Show Stats Download Patch File Download Diff File Expand all files Collapse all files

3
backend/open_webui/routers/images.py
View File

@ -333,10 +333,11 @@ def get_models(request: Request, user=Depends(get_verified_user)):
return [ return [
{"id": "dall-e-2", "name": "DALL·E 2"}, {"id": "dall-e-2", "name": "DALL·E 2"},
{"id": "dall-e-3", "name": "DALL·E 3"}, {"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": elif request.app.state.config.IMAGE_GENERATION_ENGINE == "gemini":
return [ 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": elif request.app.state.config.IMAGE_GENERATION_ENGINE == "comfyui":
# TODO - get models from comfyui # TODO - get models from comfyui