Update api/ltx_server.py
Browse files- api/ltx_server.py +13 -9
api/ltx_server.py
CHANGED
|
@@ -23,7 +23,7 @@ import traceback
|
|
| 23 |
|
| 24 |
# Singletons do projeto para VAE e Encoder
|
| 25 |
from tools.video_encode_tool import video_encode_tool_singleton
|
| 26 |
-
from
|
| 27 |
|
| 28 |
# --- 2. GERENCIAMENTO DE DEPEND脢NCIAS E SETUP ---
|
| 29 |
def _query_gpu_processes_via_nvml(device_index: int) -> List[Dict]:
|
|
@@ -113,6 +113,18 @@ if not LTX_VIDEO_REPO_DIR.exists():
|
|
| 113 |
print(f"[DEBUG] Reposit贸rio n茫o encontrado em {LTX_VIDEO_REPO_DIR}. Rodando setup...")
|
| 114 |
run_setup()
|
| 115 |
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
| 116 |
def add_deps_to_path():
|
| 117 |
repo_path = str(LTX_VIDEO_REPO_DIR.resolve())
|
| 118 |
if str(LTX_VIDEO_REPO_DIR.resolve()) not in sys.path:
|
|
@@ -122,14 +134,6 @@ def add_deps_to_path():
|
|
| 122 |
add_deps_to_path()
|
| 123 |
|
| 124 |
# --- 3. IMPORTA脟脮ES ESPEC脥FICAS DO MODELO ---
|
| 125 |
-
from .api.ltx.inference import (
|
| 126 |
-
create_ltx_video_pipeline,
|
| 127 |
-
create_latent_upsampler,
|
| 128 |
-
load_image_to_tensor_with_resize_and_crop,
|
| 129 |
-
seed_everething,
|
| 130 |
-
calculate_padding,
|
| 131 |
-
load_media_file,
|
| 132 |
-
)
|
| 133 |
from ltx_video.pipelines.pipeline_ltx_video import ConditioningItem, LTXMultiScalePipeline
|
| 134 |
from ltx_video.utils.skip_layer_strategy import SkipLayerStrategy
|
| 135 |
|
|
|
|
| 23 |
|
| 24 |
# Singletons do projeto para VAE e Encoder
|
| 25 |
from tools.video_encode_tool import video_encode_tool_singleton
|
| 26 |
+
from manager.vae_manager import vae_manager_singleton
|
| 27 |
|
| 28 |
# --- 2. GERENCIAMENTO DE DEPEND脢NCIAS E SETUP ---
|
| 29 |
def _query_gpu_processes_via_nvml(device_index: int) -> List[Dict]:
|
|
|
|
| 113 |
print(f"[DEBUG] Reposit贸rio n茫o encontrado em {LTX_VIDEO_REPO_DIR}. Rodando setup...")
|
| 114 |
run_setup()
|
| 115 |
|
| 116 |
+
|
| 117 |
+
from api.ltx.inference import (
|
| 118 |
+
create_ltx_video_pipeline,
|
| 119 |
+
create_latent_upsampler,
|
| 120 |
+
load_image_to_tensor_with_resize_and_crop,
|
| 121 |
+
seed_everething,
|
| 122 |
+
calculate_padding,
|
| 123 |
+
load_media_file,
|
| 124 |
+
)
|
| 125 |
+
|
| 126 |
+
|
| 127 |
+
|
| 128 |
def add_deps_to_path():
|
| 129 |
repo_path = str(LTX_VIDEO_REPO_DIR.resolve())
|
| 130 |
if str(LTX_VIDEO_REPO_DIR.resolve()) not in sys.path:
|
|
|
|
| 134 |
add_deps_to_path()
|
| 135 |
|
| 136 |
# --- 3. IMPORTA脟脮ES ESPEC脥FICAS DO MODELO ---
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
| 137 |
from ltx_video.pipelines.pipeline_ltx_video import ConditioningItem, LTXMultiScalePipeline
|
| 138 |
from ltx_video.utils.skip_layer_strategy import SkipLayerStrategy
|
| 139 |
|