Spaces:

LanguageBind
/

Open-Sora-Plan-v1.0.0

Runtime error

LanguageBind commited on May 27

Commit

972f2c0

•

1 Parent(s): 65ca53b

Update opensora/models/ae/videobase/losses/lpips.py

Files changed (1) hide show

opensora/models/ae/videobase/losses/lpips.py CHANGED Viewed

@@ -18,7 +18,7 @@ class LPIPS(nn.Module):
         self.lin2 = NetLinLayer(self.chns[2], use_dropout=use_dropout)
         self.lin3 = NetLinLayer(self.chns[3], use_dropout=use_dropout)
         self.lin4 = NetLinLayer(self.chns[4], use_dropout=use_dropout)
-        self.load_from_pretrained()
         for param in self.parameters():
             param.requires_grad = False

         self.lin2 = NetLinLayer(self.chns[2], use_dropout=use_dropout)
         self.lin3 = NetLinLayer(self.chns[3], use_dropout=use_dropout)
         self.lin4 = NetLinLayer(self.chns[4], use_dropout=use_dropout)
+        # self.load_from_pretrained()
         for param in self.parameters():
             param.requires_grad = False