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
|
@@ -365,8 +365,10 @@ class CausalVAEModel(VideoBaseAE_PL):
|
|
| 365 |
quant_conv_cls = resolve_str_to_obj(q_conv)
|
| 366 |
self.quant_conv = quant_conv_cls(2 * z_channels, 2 * embed_dim, 1)
|
| 367 |
self.post_quant_conv = quant_conv_cls(embed_dim, z_channels, 1)
|
|
|
|
| 368 |
if hasattr(self.loss, "discriminator"):
|
| 369 |
self.automatic_optimization = False
|
|
|
|
| 370 |
|
| 371 |
def encode(self, x):
|
| 372 |
if self.use_tiling and (
|
|
|
|
| 365 |
quant_conv_cls = resolve_str_to_obj(q_conv)
|
| 366 |
self.quant_conv = quant_conv_cls(2 * z_channels, 2 * embed_dim, 1)
|
| 367 |
self.post_quant_conv = quant_conv_cls(embed_dim, z_channels, 1)
|
| 368 |
+
'''
|
| 369 |
if hasattr(self.loss, "discriminator"):
|
| 370 |
self.automatic_optimization = False
|
| 371 |
+
'''
|
| 372 |
|
| 373 |
def encode(self, x):
|
| 374 |
if self.use_tiling and (
|