EuuIia commited on
Commit
829e1b9
·
verified ·
1 Parent(s): 441491f

Update managers/vae_manager.py

Browse files
Files changed (1) hide show
  1. managers/vae_manager.py +3 -0
managers/vae_manager.py CHANGED
@@ -4,6 +4,9 @@
4
  import torch
5
  import contextlib
6
 
 
 
 
7
  class _SimpleVAEManager:
8
  def __init__(self, pipeline=None, device=None, autocast_dtype=torch.float32):
9
  """
 
4
  import torch
5
  import contextlib
6
 
7
+ from ltx_video.models.autoencoders.vae_encode import vae_encode, vae_decode
8
+
9
+
10
  class _SimpleVAEManager:
11
  def __init__(self, pipeline=None, device=None, autocast_dtype=torch.float32):
12
  """