From 3b94e34f5eaab336a5b16c5d07365fd61f8fcdeb Mon Sep 17 00:00:00 2001 From: kijai <40791699+kijai@users.noreply.github.com> Date: Fri, 26 Apr 2024 09:41:53 +0300 Subject: [PATCH] Update hidiffusion.py --- hidiffusion/hidiffusion.py | 3 ++- 1 file changed, 2 insertions(+), 1 deletion(-) diff --git a/hidiffusion/hidiffusion.py b/hidiffusion/hidiffusion.py index 0ca802f..dfc41de 100644 --- a/hidiffusion/hidiffusion.py +++ b/hidiffusion/hidiffusion.py @@ -1763,7 +1763,8 @@ def make_diffusers_upsampler_block(block_class: Type[torch.nn.Module]) -> Type[t else: self.T1 = int(self.max_timestep * self.T1_ratio) if self.timestep < self.T1: - hidden_states = F.interpolate(hidden_states, scale_factor=2.0, mode='bicubic') + if ori_H != hidden_states.shape[2] and ori_W != hidden_states.shape[3]: + hidden_states = F.interpolate(hidden_states, scale_factor=2.0, mode='bicubic') self.timestep += 1 if self.timestep == self.max_timestep: self.timestep = 0