Reject non-positive Flux2 Klein inference steps (#3717)

Signed-off-by: MmMaiIIi <2360655509@qq.com>
This commit is contained in:
JiaHong authored and GitHub committed 2026-05-19 11:52:41 +08:00
1 parent 3ac739817f
commit ca9fd0b71c
2 files changed
+124 -1

No files matched your search

@@ -865,7 +865,11 @@ class Flux2KleinPipeline(nn.Module, CFGParallelMixin, SupportImageInput, Diffusi
height = req.sampling_params.height or height
width = req.sampling_params.width or width
num_inference_steps = req.sampling_params.num_inference_steps or num_inference_steps
num_inference_steps = (
req.sampling_params.num_inference_steps
if req.sampling_params.num_inference_steps is not None
else num_inference_steps
)
sigmas = req.sampling_params.sigmas or sigmas
guidance_scale = (
req.sampling_params.guidance_scale if req.sampling_params.guidance_scale is not None else guidance_scale