[diffusion] chore: fix default offload setting for image generation model (#15928)

This commit is contained in:
Mick
2025-12-28 20:45:33 +08:00
committed by GitHub
parent d90f9bfc4e
commit 9f8e23071a
3 changed files with 28 additions and 20 deletions
@@ -285,7 +285,7 @@ class SamplingParams:
if pipeline_config.task_type.is_image_gen():
# settle num_frames
if not server_args.pipeline_config.allow_set_num_frames():
logger.debug(f"num_frames set to 1 for image generation model")
logger.debug(f"Setting `num_frames` to 1 for image generation model")
self.num_frames = 1
elif self.adjust_frames: