OpenWebUI/open-webui-custom
1
0
Fork 0
Code Issues Pull Requests Actions Packages Projects Releases Wiki Activity

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
1 changed files with 13 additions and 4 deletions
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