From de0199c6b2f165d1ca7710f7e899629a5be4082c Mon Sep 17 00:00:00 2001 From: Shen-Chenhui Date: Tue, 16 Apr 2024 17:30:03 +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 5fdd923..9742211 100644 --- a/opensora/models/vae/vae_3d_v2.py +++ b/opensora/models/vae/vae_3d_v2.py @@ -890,7 +890,7 @@ class VEALoss(nn.Module): # KL Loss self.kl_loss_weight = kl_loss_weight # Perceptual Loss - self.perceptual_loss_fn = LPIPS().eval() + self.perceptual_loss_fn = LPIPS().eval().to(device, dtype) self.perceptual_loss_weight = perceptual_loss_weight self.logvar = nn.Parameter(torch.ones(size=()) * logvar_init)