EuuIia commited on
Commit
ff5ac95
verified
1 Parent(s): f995d54

Update managers/vae_manager.py

Browse files
Files changed (1) hide show
  1. managers/vae_manager.py +20 -0
managers/vae_manager.py CHANGED
@@ -3,6 +3,26 @@
3
 
4
  import torch
5
  import contextlib
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
6
 
7
  from ltx_video.models.autoencoders.vae_encode import vae_encode, vae_decode
8
 
 
3
 
4
  import torch
5
  import contextlib
6
+ import os
7
+ import subprocess
8
+ import sys
9
+ from pathlib import Path
10
+
11
+ DEPS_DIR = Path("/data")
12
+ LTX_VIDEO_REPO_DIR = DEPS_DIR / "LTX-Video"
13
+ if not LTX_VIDEO_REPO_DIR.exists():
14
+ print(f"[DEBUG] Reposit贸rio n茫o encontrado em {LTX_VIDEO_REPO_DIR}. Rodando setup...")
15
+ run_setup()
16
+
17
+ def add_deps_to_path():
18
+ repo_path = str(LTX_VIDEO_REPO_DIR.resolve())
19
+ if str(LTX_VIDEO_REPO_DIR.resolve()) not in sys.path:
20
+ sys.path.insert(0, repo_path)
21
+ print(f"[DEBUG] Repo adicionado ao sys.path: {repo_path}")
22
+
23
+ add_deps_to_path()
24
+
25
+
26
 
27
  from ltx_video.models.autoencoders.vae_encode import vae_encode, vae_decode
28