[diffusion] feat: add support for additional sampling parameters in video generation API (#15062)

Co-authored-by: Mick <mickjagger19@icloud.com>
This commit is contained in:
Xiaoyu Zhang
2025-12-14 19:44:03 +08:00
committed by GitHub
co-authored by Mick
parent fdfabb7afc
commit e3f51e823e
3 changed files with 35 additions and 12 deletions
@@ -123,6 +123,7 @@ class SamplingParams:
# Denoising parameters
num_inference_steps: int = None
guidance_scale: float = None
guidance_scale_2: float = None
guidance_rescale: float = 0.0
boundary_ratio: float | None = None
@@ -494,6 +495,13 @@ class SamplingParams:
default=SamplingParams.guidance_scale,
help="Classifier-free guidance scale",
)
parser.add_argument(
"--guidance-scale-2",
type=float,
default=SamplingParams.guidance_scale_2,
dest="guidance_scale_2",
help="Secondary guidance scale for dual-guidance models (e.g., Wan low-noise expert)",
)
parser.add_argument(
"--guidance-rescale",
type=float,
@@ -589,7 +597,7 @@ class SamplingParams:
attrs = sampling_params_fields & args_attrs
args.height_not_provided = False
args.width_not_provided = False
return {attr: getattr(args, attr) for attr in attrs}
return {attr: getattr(args, attr) for attr in attrs if hasattr(args, attr)}
def output_file_path(self):
return os.path.join(self.output_path, self.output_file_name)