support height width checker

2026-03-24 01:48:13 +00:00 · 2024-11-13 12:39:09 +08:00
parent 7f4ba62d4f
commit e17ad83fb5
10 changed files with 20 additions and 0 deletions
--- 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)