AI/diffusers
1
0
Fork 0
mirror of https://github.com/huggingface/diffusers.git synced 2026-01-27 17:22:53 +03:00
Code Activity

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

Browse Source 
contiguous tensors in resnet

Co-authored-by: YiYi Xu <yixu310@gmail.com>
This commit is contained in:
Aryan
2024-12-20 11:45:37 +05:30
committed by GitHub
parent 41ba8c0bf6
commit 151b74cd77
1 changed files with 1 additions and 0 deletions
Download Patch File Download Diff File Expand all files Collapse all files

1
src/diffusers/models/autoencoders/autoencoder_kl_hunyuan_video.py
View File

@@ -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)