Spaces:
Runtime error
Runtime error
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,
|