nicolabs/Open-Sora
1
0
Fork 0
mirror of https://github.com/hpcaitech/Open-Sora.git synced 2026-04-11 05:13:31 +02:00
Code Issues Projects Releases Packages Wiki Activity Actions 1

debug

Browse source
This commit is contained in:
Shen-Chenhui 2024-04-16 17:30:03 +08:00
parent dc45231c8a
commit de0199c6b2
1 changed files with 1 additions and 1 deletions
Show stats Download patch file Download diff file Expand all files Collapse all files

2
opensora/models/vae/vae_3d_v2.py
View file

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