scriptoriumadmin/open-webui
1
0
Fork 0
mirror of https://github.com/open-webui/open-webui.git synced 2026-09-06 08:18:53 +00:00
Code Issues Projects Releases Packages Wiki Activity Actions 10

feat: add support for passing 'steps' in image generation requests (#20337)

Browse source
This commit is contained in:
G30 2026-01-03 09:11:13 -05:00 • committed by GitHub
parent cfecca8eda
commit bf42535d12
No known key found for this signature in database
GPG key ID: B5690EEEBB952194
1 changed files with 13 additions and 4 deletions
Show stats Download patch file Download diff file Expand all files Collapse all files

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

@ -463,6 +463,7 @@ class CreateImageForm(BaseModel):
prompt: str
size: Optional[str] = None
n: int = 1
steps: Optional[int] = None
negative_prompt: Optional[str] = None
@ -706,8 +707,12 @@ async def image_generations(
"n": form_data.n,
}
if request.app.state.config.IMAGE_STEPS is not None:
data["steps"] = request.app.state.config.IMAGE_STEPS
if request.app.state.config.IMAGE_STEPS is not None or form_data.steps is not None:
data["steps"] = (
form_data.steps
if form_data.steps is not None
else request.app.state.config.IMAGE_STEPS
)
if form_data.negative_prompt is not None:
data["negative_prompt"] = form_data.negative_prompt
@ -765,8 +770,12 @@ async def image_generations(
"height": height,
}
if request.app.state.config.IMAGE_STEPS is not None:
data["steps"] = request.app.state.config.IMAGE_STEPS
if request.app.state.config.IMAGE_STEPS is not None or form_data.steps is not None:
data["steps"] = (
form_data.steps
if form_data.steps is not None
else request.app.state.config.IMAGE_STEPS
)
if form_data.negative_prompt is not None:
data["negative_prompt"] = form_data.negative_prompt