Spaces:
Runtime error
Runtime error
LanguageBind
commited on
Update opensora/models/ae/videobase/losses/lpips.py
Browse files
opensora/models/ae/videobase/losses/lpips.py
CHANGED
@@ -18,7 +18,7 @@ class LPIPS(nn.Module):
|
|
18 |
self.lin2 = NetLinLayer(self.chns[2], use_dropout=use_dropout)
|
19 |
self.lin3 = NetLinLayer(self.chns[3], use_dropout=use_dropout)
|
20 |
self.lin4 = NetLinLayer(self.chns[4], use_dropout=use_dropout)
|
21 |
-
self.load_from_pretrained()
|
22 |
for param in self.parameters():
|
23 |
param.requires_grad = False
|
24 |
|
|
|
18 |
self.lin2 = NetLinLayer(self.chns[2], use_dropout=use_dropout)
|
19 |
self.lin3 = NetLinLayer(self.chns[3], use_dropout=use_dropout)
|
20 |
self.lin4 = NetLinLayer(self.chns[4], use_dropout=use_dropout)
|
21 |
+
# self.load_from_pretrained()
|
22 |
for param in self.parameters():
|
23 |
param.requires_grad = False
|
24 |
|