Spaces:
Runtime error
Runtime error
LanguageBind
commited on
Commit
•
45a109e
1
Parent(s):
810fa8c
Update opensora/models/ae/videobase/causal_vae/modeling_causalvae.py
Browse files
opensora/models/ae/videobase/causal_vae/modeling_causalvae.py
CHANGED
@@ -322,11 +322,11 @@ class CausalVAEModel(VideoBaseAE_PL):
|
|
322 |
|
323 |
self.learning_rate = lr
|
324 |
self.lr_g_factor = 1.0
|
325 |
-
|
326 |
self.loss = resolve_str_to_obj(loss_type, append=False)(
|
327 |
**loss_params
|
328 |
)
|
329 |
-
|
330 |
self.encoder = Encoder(
|
331 |
z_channels=z_channels,
|
332 |
hidden_size=hidden_size,
|
|
|
322 |
|
323 |
self.learning_rate = lr
|
324 |
self.lr_g_factor = 1.0
|
325 |
+
'''
|
326 |
self.loss = resolve_str_to_obj(loss_type, append=False)(
|
327 |
**loss_params
|
328 |
)
|
329 |
+
'''
|
330 |
self.encoder = Encoder(
|
331 |
z_channels=z_channels,
|
332 |
hidden_size=hidden_size,
|