From 119653c3f2dee1dc278f7e2cf9a247db74a1e405 Mon Sep 17 00:00:00 2001 From: Jaret Burkett Date: Mon, 25 Aug 2025 10:33:28 -0600 Subject: [PATCH] Force width, height, and num frames to always be the proper sizes for Wan 2.2 models --- .../diffusion_models/wan22/wan22_pipeline.py | 8 ++++++++ 1 file changed, 8 insertions(+) diff --git a/extensions_built_in/diffusion_models/wan22/wan22_pipeline.py b/extensions_built_in/diffusion_models/wan22/wan22_pipeline.py index f434327f..8d7db242 100644 --- a/extensions_built_in/diffusion_models/wan22/wan22_pipeline.py +++ b/extensions_built_in/diffusion_models/wan22/wan22_pipeline.py @@ -73,6 +73,14 @@ class Wan22Pipeline(WanPipeline): if isinstance(callback_on_step_end, (PipelineCallback, MultiPipelineCallbacks)): callback_on_step_end_tensor_inputs = callback_on_step_end.tensor_inputs + + if num_frames % self.vae_scale_factor_temporal != 1: + num_frames = num_frames // self.vae_scale_factor_temporal * self.vae_scale_factor_temporal + 1 + num_frames = max(num_frames, 1) + + + width = width // (self.vae.config.scale_factor_spatial * 2) * (self.vae.config.scale_factor_spatial * 2) + height = height // (self.vae.config.scale_factor_spatial * 2) * (self.vae.config.scale_factor_spatial * 2) # unload vae and transformer vae_device = self.vae.device