Spaces:
Paused
Paused
Update api/ltx/ltx_aduc_pipeline.py
Browse files
api/ltx/ltx_aduc_pipeline.py
CHANGED
|
@@ -46,7 +46,7 @@ try:
|
|
| 46 |
from api.ltx.vae_aduc_pipeline import vae_ltx_aduc_pipeline
|
| 47 |
from tools.video_encode_tool import video_encode_tool_singleton
|
| 48 |
from api.ltx.ltx_utils import build_ltx_pipeline_on_cpu, seed_everything
|
| 49 |
-
from api.ltx.ltx_aduc_manager import LatentConditioningItem
|
| 50 |
from utils.debug_utils import log_function_io
|
| 51 |
except ImportError as e:
|
| 52 |
logging.critical(f"A crucial import from the local API/architecture failed. Error: {e}", exc_info=True)
|
|
@@ -72,7 +72,7 @@ class LtxAducPipeline:
|
|
| 72 |
target_vae_device_str = str(gpu_manager.get_ltx_vae_device())
|
| 73 |
logging.info(f"LTX allocated to devices: Main='{target_main_device_str}', VAE='{target_vae_device_str}'")
|
| 74 |
|
| 75 |
-
self.config =
|
| 76 |
self._resolve_model_paths_from_cache()
|
| 77 |
|
| 78 |
self.pipeline, self.latent_upsampler = build_ltx_pipeline_on_cpu(self.config)
|
|
|
|
| 46 |
from api.ltx.vae_aduc_pipeline import vae_ltx_aduc_pipeline
|
| 47 |
from tools.video_encode_tool import video_encode_tool_singleton
|
| 48 |
from api.ltx.ltx_utils import build_ltx_pipeline_on_cpu, seed_everything
|
| 49 |
+
from api.ltx.ltx_aduc_manager import LatentConditioningItem, ltx_aduc_manager
|
| 50 |
from utils.debug_utils import log_function_io
|
| 51 |
except ImportError as e:
|
| 52 |
logging.critical(f"A crucial import from the local API/architecture failed. Error: {e}", exc_info=True)
|
|
|
|
| 72 |
target_vae_device_str = str(gpu_manager.get_ltx_vae_device())
|
| 73 |
logging.info(f"LTX allocated to devices: Main='{target_main_device_str}', VAE='{target_vae_device_str}'")
|
| 74 |
|
| 75 |
+
self.config = ltx_aduc_manager.load_config()
|
| 76 |
self._resolve_model_paths_from_cache()
|
| 77 |
|
| 78 |
self.pipeline, self.latent_upsampler = build_ltx_pipeline_on_cpu(self.config)
|