Spaces:
Paused
Paused
Update api/ltx/ltx_aduc_manager.py
Browse files
api/ltx/ltx_aduc_manager.py
CHANGED
|
@@ -145,14 +145,14 @@ class LTXAducManager:
|
|
| 145 |
main_device = gpu_manager.get_ltx_device()
|
| 146 |
vae_device = gpu_manager.get_ltx_vae_device()
|
| 147 |
# Em uma arquitetura futura, poderíamos ter múltiplos workers. Por enquanto, temos um.
|
| 148 |
-
self.worker = LTXWorker(str(main_device), str(vae_device), load_config())
|
| 149 |
|
| 150 |
-
def load_config() -> Dict:
|
| 151 |
"""Loads the YAML configuration file."""
|
| 152 |
config_path = LTX_VIDEO_REPO_DIR / "configs" / "ltxv-13b-0.9.8-distilled-fp8.yaml"
|
| 153 |
with open(config_path, "r") as file:
|
| 154 |
return yaml.safe_load(file)
|
| 155 |
-
|
| 156 |
def get_pipeline(self) -> LTXVideoPipeline:
|
| 157 |
return self.worker.pipeline
|
| 158 |
|
|
|
|
| 145 |
main_device = gpu_manager.get_ltx_device()
|
| 146 |
vae_device = gpu_manager.get_ltx_vae_device()
|
| 147 |
# Em uma arquitetura futura, poderíamos ter múltiplos workers. Por enquanto, temos um.
|
| 148 |
+
self.worker = LTXWorker(str(main_device), str(vae_device), self.load_config())
|
| 149 |
|
| 150 |
+
def load_config(self) -> Dict:
|
| 151 |
"""Loads the YAML configuration file."""
|
| 152 |
config_path = LTX_VIDEO_REPO_DIR / "configs" / "ltxv-13b-0.9.8-distilled-fp8.yaml"
|
| 153 |
with open(config_path, "r") as file:
|
| 154 |
return yaml.safe_load(file)
|
| 155 |
+
|
| 156 |
def get_pipeline(self) -> LTXVideoPipeline:
|
| 157 |
return self.worker.pipeline
|
| 158 |
|