Unverified Commit 151b74cd authored by Aryan's avatar Aryan Committed by GitHub
Browse files

Make tensors in ResNet contiguous for Hunyuan VAE (#10309)



contiguous tensors in resnet
Co-authored-by: default avatarYiYi Xu <yixu310@gmail.com>
parent 41ba8c0b
...@@ -168,6 +168,7 @@ class HunyuanVideoResnetBlockCausal3D(nn.Module): ...@@ -168,6 +168,7 @@ class HunyuanVideoResnetBlockCausal3D(nn.Module):
self.conv_shortcut = HunyuanVideoCausalConv3d(in_channels, out_channels, 1, 1, 0) self.conv_shortcut = HunyuanVideoCausalConv3d(in_channels, out_channels, 1, 1, 0)
def forward(self, hidden_states: torch.Tensor) -> torch.Tensor: def forward(self, hidden_states: torch.Tensor) -> torch.Tensor:
hidden_states = hidden_states.contiguous()
residual = hidden_states residual = hidden_states
hidden_states = self.norm1(hidden_states) hidden_states = self.norm1(hidden_states)
......
Markdown is supported
0% or .
You are about to add 0 people to the discussion. Proceed with caution.
Finish editing this message first!
Please register or to comment