[diffusion] chore: fix default offload setting for image generation model (#15928)
This commit is contained in:
@@ -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:
|
||||
|
||||
Reference in New Issue
Block a user