Update api/seedvr/seedvr_aduc_pipeline.py

api/seedvr/seedvr_aduc_pipeline.py

@@ -17,14 +17,14 @@ import logging
 # --- IMPORTAÇÃO DOS MÓDulos Compartilhados ---
 # ==============================================================================
 try:
-    from api.gpu_manager import gpu_manager
-    from api.ltx_server_refactored_complete import video_generation_service
-    from api.utils.debug_utils import log_function_io
+    from manager.gpu_manager import gpu_manager
+    from api.ltx.ltx_server_refactored_complete import ltx_aduc_pipeline
+    from utils.debug_utils import log_function_io
 except ImportError:
     # Fallback para o decorador caso o import falhe
     def log_function_io(func):
         return func
-    logging.critical("CRITICAL: Failed to import shared modules like gpu_manager or video_generation_service.", exc_info=True)
+    logging.critical("CRITICAL: Failed to import shared modules like gpu_manager or ltx_aduc_pipeline.", exc_info=True)
     # Em um cenário real, poderíamos querer sair aqui ou desativar o servidor.
     # Por enquanto, a aplicação pode tentar continuar sem o SeedVR.
     raise
@@ -151,7 +151,7 @@ class SeedAducPipeline:
         if gpu_manager.requires_memory_swap():
             logging.warning("[SWAP] Memory swapping is active. Moving LTX service to CPU to free VRAM for SeedVR.")
             if progress: progress(0.02, "🔄 Freeing VRAM for SeedVR...")
-            video_generation_service.move_to_cpu()
+            ltx_aduc_pipeline.move_to_cpu()
 
         try:
             if progress: progress(0.05, "🎬 Extracting frames from video...")
@@ -211,7 +211,7 @@ class SeedAducPipeline:
                 ltx_main_device = gpu_manager.get_ltx_device()
                 ltx_vae_device = gpu_manager.get_ltx_vae_device()
                 # Chama a função move_to_device com os dois dispositivos
-                video_generation_service.move_to_device(
+                ltx_aduc_pipeline.move_to_device(
                     main_device_str=str(ltx_main_device),
                     vae_device_str=str(ltx_vae_device)
                 )