Create ltx_upscaler_manager_helpers.py

ltx_upscaler_manager_helpers.py

@@ -1,4 +1,4 @@
-# ltx_upscaler_manager_helpers.py
+# FILE: ltx_upscaler_manager_helpers.py
 # Gerente de Pool para o revezamento de workers de Upscaling.
 # Este arquivo é parte do projeto Euia-AducSdr e está sob a licença AGPL v3.
 # Copyright (C) 4 de Agosto de 2025  Carlos Rodrigues dos Santos
@@ -14,7 +14,7 @@ class LtxUpscalerPoolManager:
     Gerencia um pool de LtxUpscalerWorkers, orquestrando um revezamento entre GPUs
     para a tarefa de upscaling.
     """
-    def __init__(self, device_ids=['cuda:2', 'cuda:3']):
+    def __init__(self, device_ids=['cuda:0', 'cuda:1']):
         print(f"LTX UPSCALER POOL MANAGER: Criando workers para os dispositivos: {device_ids}")
         self.workers = [LtxUpscaler(device_id) for device_id in device_ids]
         self.current_worker_index = 0
@@ -53,10 +53,10 @@ class LtxUpscalerPoolManager:
             print(f"UPSCALER POOL MANAGER: Worker em {worker_to_use.device} iniciando upscale de {os.path.basename(video_path_low_res)}...")
             worker_to_use.upscale_video_fragment(video_path_low_res, output_path, video_fps)
             print(f"UPSCALER POOL MANAGER: Upscale de {os.path.basename(video_path_low_res)} concluído.")
-
+            return output_path
         finally:
             # A limpeza do worker_to_use será feita na próxima chamada
             pass
 
 # --- Instância Singleton do Gerenciador de Upscaling ---
-ltx_upscaler_manager_singleton = LtxUpscalerPoolManager(device_ids=['cuda:2', 'cuda:3'])
+ltx_upscaler_manager_singleton = LtxUpscalerPoolManager(device_ids=['cuda:0', 'cuda:1'])