From 1edc7c60ede4bbd43d7c708699f05f4e6c838574 Mon Sep 17 00:00:00 2001 From: Shen-Chenhui Date: Fri, 29 Mar 2024 10:17:09 +0800 Subject: [PATCH] debug --- opensora/models/vae/vae_3d.py | 2 +- 1 file changed, 1 insertion(+), 1 deletion(-) diff --git a/opensora/models/vae/vae_3d.py b/opensora/models/vae/vae_3d.py index 47a6cd9..a1f233e 100644 --- a/opensora/models/vae/vae_3d.py +++ b/opensora/models/vae/vae_3d.py @@ -42,7 +42,7 @@ class ResBlock(nn.Module): # SCH: MAGVIT uses GroupNorm by default self.norm1 = nn.GroupNorm(num_groups, in_out_channels) self.conv1 = conv_fn(in_out_channels, self.filters, kernel_size=(3, 3, 3), bias=False) - self.norm2 = nn.GroupNorm(num_groups, self.filters) + self.norm2 = nn.GroupNorm(num_groups, self.filters, device=device) self.conv2 = conv_fn(self.filters, self.filters, kernel_size=(3, 3, 3), bias=False) if self.use_conv_shortcut: self.conv3 = conv_fn(in_out_channels, self.filters, kernel_size=(3, 3, 3), bias=False)