Merge pull request #265 from modelscope/dev

support height width checker

diffsynth/pipelines/cog_video.py

@@ -77,6 +77,8 @@ class CogVideoPipeline(BasePipeline):
         progress_bar_cmd=tqdm,
         progress_bar_st=None,
     ):
+        height, width = self.check_resize_height_width(height, width)
+        
         # Tiler parameters
         tiler_kwargs = {"tiled": tiled, "tile_size": tile_size, "tile_stride": tile_stride}
 

diffsynth/pipelines/flux_image.py

@@ -156,6 +156,8 @@ class FluxImagePipeline(BasePipeline):
         progress_bar_cmd=tqdm,
         progress_bar_st=None,
     ):
+        height, width = self.check_resize_height_width(height, width)
+        
         # Tiler parameters
         tiler_kwargs = {"tiled": tiled, "tile_size": tile_size, "tile_stride": tile_stride}
 

diffsynth/pipelines/hunyuan_image.py

@@ -230,6 +230,8 @@ class HunyuanDiTImagePipeline(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)
 

diffsynth/pipelines/omnigen_image.py

@@ -215,6 +215,8 @@ class OmnigenImagePipeline(BasePipeline):
         progress_bar_cmd=tqdm,
         progress_bar_st=None,
     ):
+        height, width = self.check_resize_height_width(height, width)
+        
         # Tiler parameters
         tiler_kwargs = {"tiled": tiled, "tile_size": tile_size, "tile_stride": tile_stride}
 

diffsynth/pipelines/sd3_image.py

@@ -92,6 +92,8 @@ class SD3ImagePipeline(BasePipeline):
         progress_bar_cmd=tqdm,
         progress_bar_st=None,
     ):
+        height, width = self.check_resize_height_width(height, width)
+        
         # Tiler parameters
         tiler_kwargs = {"tiled": tiled, "tile_size": tile_size, "tile_stride": tile_stride}
 

diffsynth/pipelines/sd_image.py

@@ -112,6 +112,8 @@ class SDImagePipeline(BasePipeline):
         progress_bar_cmd=tqdm,
         progress_bar_st=None,
     ):
+        height, width = self.check_resize_height_width(height, width)
+        
         # Tiler parameters
         tiler_kwargs = {"tiled": tiled, "tile_size": tile_size, "tile_stride": tile_stride}
 

diffsynth/pipelines/sd_video.py

@@ -170,6 +170,8 @@ class SDVideoPipeline(SDImagePipeline):
         progress_bar_cmd=tqdm,
         progress_bar_st=None,
     ):
+        height, width = self.check_resize_height_width(height, width)
+        
         # Tiler parameters, batch size ...
         tiler_kwargs = {"tiled": tiled, "tile_size": tile_size, "tile_stride": tile_stride}
         other_kwargs = {

diffsynth/pipelines/sdxl_image.py

@@ -135,6 +135,8 @@ class SDXLImagePipeline(BasePipeline):
         progress_bar_cmd=tqdm,
         progress_bar_st=None,
     ):
+        height, width = self.check_resize_height_width(height, width)
+        
         # Tiler parameters
         tiler_kwargs = {"tiled": tiled, "tile_size": tile_size, "tile_stride": tile_stride}
 

diffsynth/pipelines/sdxl_video.py

@@ -124,6 +124,8 @@ class SDXLVideoPipeline(SDXLImagePipeline):
         progress_bar_cmd=tqdm,
         progress_bar_st=None,
     ):
+        height, width = self.check_resize_height_width(height, width)
+        
         # Tiler parameters, batch size ...
         tiler_kwargs = {"tiled": tiled, "tile_size": tile_size, "tile_stride": tile_stride}
 

diffsynth/pipelines/svd_video.py

@@ -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)