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,