From 60892cd0900ffe71fbc90bf00f31becae3a6827c Mon Sep 17 00:00:00 2001 From: Shen-Chenhui Date: Thu, 18 Apr 2024 15:11:25 +0800 Subject: [PATCH] debug --- opensora/models/vae/vae_3d_v2.py | 2 +- 1 file changed, 1 insertion(+), 1 deletion(-) diff --git a/opensora/models/vae/vae_3d_v2.py b/opensora/models/vae/vae_3d_v2.py index bf81b89..d32b44b 100644 --- a/opensora/models/vae/vae_3d_v2.py +++ b/opensora/models/vae/vae_3d_v2.py @@ -429,7 +429,7 @@ class StyleGANDiscriminatorBlur(nn.Module): prev_filters = filters # update in_channels self.conv2 = nn.Conv3d(prev_filters, prev_filters, (3,3,3), padding=1, device=device, dtype=dtype) # NOTE: init to xavier_uniform - torch.nn.init.xavier_uniform(self.conv2.weight) + # torch.nn.init.xavier_uniform_(self.conv2.weight) self.norm1 = nn.GroupNorm(num_groups, prev_filters, dtype=dtype, device=device)