Update hidiffusion.py

2024-04-26 09:41:53 +03:00
parent 78c04fc3c5
commit 3b94e34f5e
1 changed files with 2 additions and 1 deletions
@@ -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