[diffusion] Support I2I/TI2I/I2V/TI2V warmup && T2I/T2V warmup bug fix (#16922)

Co-authored-by: yhyang201 <yhyang201@gmail.com>
Co-authored-by: gemini-code-assist[bot] <176961590+gemini-code-assist[bot]@users.noreply.github.com>
This commit is contained in:
HuangJi
2026-01-12 19:40:37 +08:00
committed by GitHub
co-authored by yhyang201 gemini-code-assist[bot] <176961590+gemini-code-assist[bot]@users.noreply.github.com>
parent aab640c99f
commit 2dadf63562
6 changed files with 113 additions and 30 deletions
@@ -100,11 +100,11 @@ class SamplingParams:
# Batch info
num_outputs_per_prompt: int = 1
seed: int = 1024
seed: int = 42
generator_device: str = "cuda" # Device for random generator: "cuda" or "cpu"
# Original dimensions (before VAE scaling)
num_frames: int = 125
num_frames: int = 1 # Default for image models
num_frames_round_down: bool = (
False # Whether to round down num_frames if it's not divisible by num_gpus
)
@@ -122,7 +122,7 @@ class SamplingParams:
# Denoising parameters
num_inference_steps: int = None
guidance_scale: float = None
guidance_scale: float = 1.0
guidance_scale_2: float = None
true_cfg_scale: float = None # for CFG vs guidance distillation (e.g., QwenImage)
guidance_rescale: float = 0.0