eeuuia commited on
Commit
81d098e
·
verified ·
1 Parent(s): 6e680c2

Update api/ltx/ltx_aduc_manager.py

Browse files
Files changed (1) hide show
  1. api/ltx/ltx_aduc_manager.py +7 -6
api/ltx/ltx_aduc_manager.py CHANGED
@@ -138,16 +138,17 @@ class LTXWorker:
138
 
139
 
140
  class LTXAducManager:
141
- # (Padrão Singleton, similar ao VincePoolManager)
142
- # ...
143
-
144
  def __init__(self):
145
- # ...
146
  main_device = gpu_manager.get_ltx_device()
147
  vae_device = gpu_manager.get_ltx_vae_device()
148
  # Em uma arquitetura futura, poderíamos ter múltiplos workers. Por enquanto, temos um.
149
- self.worker = LTXWorker(str(main_device), str(vae_device), self._load_config())
150
- # ...
 
 
 
 
 
151
 
152
  def get_pipeline(self) -> LTXVideoPipeline:
153
  return self.worker.pipeline
 
138
 
139
 
140
  class LTXAducManager:
 
 
 
141
  def __init__(self):
 
142
  main_device = gpu_manager.get_ltx_device()
143
  vae_device = gpu_manager.get_ltx_vae_device()
144
  # Em uma arquitetura futura, poderíamos ter múltiplos workers. Por enquanto, temos um.
145
+ self.worker = LTXWorker(str(main_device), str(vae_device), self.load_config())
146
+
147
+ def load_config(self) -> Dict:
148
+ """Loads the YAML configuration file."""
149
+ config_path = LTX_VIDEO_REPO_DIR / "configs" / "ltxv-13b-0.9.8-distilled-fp8.yaml"
150
+ with open(config_path, "r") as file:
151
+ return yaml.safe_load(file)
152
 
153
  def get_pipeline(self) -> LTXVideoPipeline:
154
  return self.worker.pipeline