Merge pull request #50 from maedtb/patch-1

Fix error when using SkyReels I2V Hunyuan Model
2025-02-20 16:07:41 +08:00
parent 61636b6503 3597bb87aa
commit be7222aeae
1 changed files with 1 additions and 1 deletions
@@ -296,7 +296,7 @@ def teacache_hunyuanvideo_forward(
            shape[i] = shape[i] // self.patch_size[i]
        img = img.reshape([img.shape[0]] + shape + [self.out_channels] + self.patch_size)
        img = img.permute(0, 4, 1, 5, 2, 6, 3, 7)
-        img = img.reshape(initial_shape)
+        img = img.reshape(initial_shape[0], self.out_channels, initial_shape[2], initial_shape[3], initial_shape[4])
        return img

 def teacache_ltxvmodel_forward(