From 151b74cd7758df590c523230a86230ba3bbc786f Mon Sep 17 00:00:00 2001 From: Aryan Date: Fri, 20 Dec 2024 11:45:37 +0530 Subject: [PATCH] Make tensors in ResNet contiguous for Hunyuan VAE (#10309) contiguous tensors in resnet Co-authored-by: YiYi Xu --- .../models/autoencoders/autoencoder_kl_hunyuan_video.py | 1 + 1 file changed, 1 insertion(+) diff --git a/src/diffusers/models/autoencoders/autoencoder_kl_hunyuan_video.py b/src/diffusers/models/autoencoders/autoencoder_kl_hunyuan_video.py index 5c1d94d4e1..e2236a7f20 100644 --- a/src/diffusers/models/autoencoders/autoencoder_kl_hunyuan_video.py +++ b/src/diffusers/models/autoencoders/autoencoder_kl_hunyuan_video.py @@ -168,6 +168,7 @@ class HunyuanVideoResnetBlockCausal3D(nn.Module): self.conv_shortcut = HunyuanVideoCausalConv3d(in_channels, out_channels, 1, 1, 0) def forward(self, hidden_states: torch.Tensor) -> torch.Tensor: + hidden_states = hidden_states.contiguous() residual = hidden_states hidden_states = self.norm1(hidden_states)