ti.sil/open-webui
1
0
Fork 0
mirror of https://github.com/open-webui/open-webui.git synced 2025-12-15 13:55:19 +00:00
Code Issues Projects Releases Packages Wiki Activity Actions 4

refac

Browse source
This commit is contained in:
Timothy Jaeryang Baek 2025-09-24 07:17:41 -05:00
parent 6cdadc73f2
commit 199f371f58
1 changed files with 3 additions and 7 deletions
Show stats Download patch file Download diff file Expand all files Collapse all files

10
backend/open_webui/routers/images.py
View file

@ -514,6 +514,7 @@ async def image_generations(
size = form_data.size size = form_data.size
width, height = tuple(map(int, size.split("x"))) width, height = tuple(map(int, size.split("x")))
model = get_image_model(request)
r = None r = None
try: try:
@ -531,11 +532,7 @@ async def image_generations(
headers["X-OpenWebUI-User-Role"] = user.role headers["X-OpenWebUI-User-Role"] = user.role
data = { data = {
"model": ( "model": model,
request.app.state.config.IMAGE_GENERATION_MODEL
if request.app.state.config.IMAGE_GENERATION_MODEL != ""
else "dall-e-2"
),
"prompt": form_data.prompt, "prompt": form_data.prompt,
"n": form_data.n, "n": form_data.n,
"size": ( "size": (
@ -584,7 +581,6 @@ async def image_generations(
headers["Content-Type"] = "application/json" headers["Content-Type"] = "application/json"
headers["x-goog-api-key"] = request.app.state.config.IMAGES_GEMINI_API_KEY headers["x-goog-api-key"] = request.app.state.config.IMAGES_GEMINI_API_KEY
model = get_image_model(request)
data = { data = {
"instances": {"prompt": form_data.prompt}, "instances": {"prompt": form_data.prompt},
"parameters": { "parameters": {
@ -640,7 +636,7 @@ async def image_generations(
} }
) )
res = await comfyui_generate_image( res = await comfyui_generate_image(
request.app.state.config.IMAGE_GENERATION_MODEL, model,
form_data, form_data,
user.id, user.id,
request.app.state.config.COMFYUI_BASE_URL, request.app.state.config.COMFYUI_BASE_URL,