[diffusion] fix: pass progressive params through image API (#30148)

2026-07-06 14:45:45 +08:00
parent 6bb2918938
commit de00b838c4
2 changed files with 19 additions and 0 deletions
@@ -201,6 +201,21 @@ async def generations(
            perf_dump_path=request.perf_dump_path,
            use_pe=_get_extra_field(request, "use_pe"),
            preset=_get_extra_field(request, "preset"),
+            progressive_mode=(
+                request.progressive_mode
+                if request.progressive_mode is not None
+                else _get_extra_field(request, "progressive_mode")
+            ),
+            progressive_levels=(
+                request.progressive_levels
+                if request.progressive_levels is not None
+                else _get_extra_field(request, "progressive_levels")
+            ),
+            progressive_delta=(
+                request.progressive_delta
+                if request.progressive_delta is not None
+                else _get_extra_field(request, "progressive_delta")
+            ),
        )
        trace_headers = extract_trace_headers(raw_request.headers)
        batch = prepare_request(
@@ -59,6 +59,10 @@ class ImageGenerationsRequest(BaseModel):
    diffusers_kwargs: Optional[Dict[str, Any]] = None  # kwargs for diffusers backend
    # Performance profiling
    perf_dump_path: Optional[str] = None
+    # Progressive resolution generation
+    progressive_mode: Optional[str] = None
+    progressive_levels: Optional[int] = None
+    progressive_delta: Optional[float] = None


 # Video API protocol models