support height width checker

This commit is contained in:
Artiprocher
2024-11-13 12:39:09 +08:00
parent 7f4ba62d4f
commit e17ad83fb5
10 changed files with 20 additions and 0 deletions

View File

@@ -130,6 +130,8 @@ class SVDVideoPipeline(BasePipeline):
progress_bar_cmd=tqdm,
progress_bar_st=None,
):
height, width = self.check_resize_height_width(height, width)
# Prepare scheduler
self.scheduler.set_timesteps(num_inference_steps, denoising_strength=denoising_strength)