diff --git a/diffsynth/pipelines/cog_video.py b/diffsynth/pipelines/cog_video.py index 959038f..4b7f336 100644 --- a/diffsynth/pipelines/cog_video.py +++ b/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} diff --git a/diffsynth/pipelines/flux_image.py b/diffsynth/pipelines/flux_image.py index 40488dc..69b7a16 100644 --- a/diffsynth/pipelines/flux_image.py +++ b/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} diff --git a/diffsynth/pipelines/hunyuan_image.py b/diffsynth/pipelines/hunyuan_image.py index 9c6736d..0c6f6d5 100644 --- a/diffsynth/pipelines/hunyuan_image.py +++ b/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) diff --git a/diffsynth/pipelines/omnigen_image.py b/diffsynth/pipelines/omnigen_image.py index 89fd4ac..428e7f0 100644 --- a/diffsynth/pipelines/omnigen_image.py +++ b/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} diff --git a/diffsynth/pipelines/sd3_image.py b/diffsynth/pipelines/sd3_image.py index 97d388d..c609873 100644 --- a/diffsynth/pipelines/sd3_image.py +++ b/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} diff --git a/diffsynth/pipelines/sd_image.py b/diffsynth/pipelines/sd_image.py index 2c7821e..c22c3fe 100644 --- a/diffsynth/pipelines/sd_image.py +++ b/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} diff --git a/diffsynth/pipelines/sd_video.py b/diffsynth/pipelines/sd_video.py index e9be94a..4337beb 100644 --- a/diffsynth/pipelines/sd_video.py +++ b/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 = { diff --git a/diffsynth/pipelines/sdxl_image.py b/diffsynth/pipelines/sdxl_image.py index 9b522ba..499c4bb 100644 --- a/diffsynth/pipelines/sdxl_image.py +++ b/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} diff --git a/diffsynth/pipelines/sdxl_video.py b/diffsynth/pipelines/sdxl_video.py index cfcccd5..308590c 100644 --- a/diffsynth/pipelines/sdxl_video.py +++ b/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} diff --git a/diffsynth/pipelines/svd_video.py b/diffsynth/pipelines/svd_video.py index 2dc6311..b71597e 100644 --- a/diffsynth/pipelines/svd_video.py +++ b/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)