eeuuia commited on
Commit
518ff71
·
verified ·
1 Parent(s): c8f8a7f

Update api/ltx/ltx_aduc_manager.py

Browse files
Files changed (1) hide show
  1. api/ltx/ltx_aduc_manager.py +3 -3
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