Update api/seedvr/seedvr_aduc_pipeline.py

api/seedvr/seedvr_aduc_pipeline.py

@@ -18,7 +18,7 @@ import logging
 # ==============================================================================
 try:
     from managers.gpu_manager import gpu_manager
-    from api.ltx.ltx_server_refactored_complete import ltx_aduc_pipeline
+    #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
@@ -148,10 +148,10 @@ class SeedAducPipeline:
         """
         if progress: progress(0.01, "⌛ Initializing SeedVR inference...")
 
-        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...")
-            ltx_aduc_pipeline.move_to_cpu()
+        #if false: #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...")
+        #    ltx_aduc_pipeline.move_to_cpu()
 
         try:
             if progress: progress(0.05, "🎬 Extracting frames from video...")
@@ -202,20 +202,20 @@ class SeedAducPipeline:
             logging.info(f"✅ Video successfully saved to: {output_filepath}")
             return str(output_filepath)
 
-        finally:
+        #finally:
             # --- CORREÇÃO IMPORTANTE ---
             # Restaura o LTX para seus dispositivos corretos (main e vae)
-            if gpu_manager.requires_memory_swap():
-                logging.warning("[SWAP] SeedVR inference finished. Moving LTX service back to GPU(s)...")
-                if progress: progress(0.99, "🔄 Restoring LTX environment...")
-                ltx_main_device = gpu_manager.get_ltx_device()
-                ltx_vae_device = gpu_manager.get_ltx_vae_device()
+            #if gpu_manager.requires_memory_swap():
+                #logging.warning("[SWAP] SeedVR inference finished. Moving LTX service back to GPU(s)...")
+                #if progress: progress(0.99, "🔄 Restoring LTX environment...")
+                #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
-                ltx_aduc_pipeline.move_to_device(
-                    main_device_str=str(ltx_main_device),
-                    vae_device_str=str(ltx_vae_device)
-                )
-                logging.info(f"[SWAP] LTX service restored to Main: {ltx_main_device}, VAE: {ltx_vae_device}.")
+                #ltx_aduc_pipeline.move_to_device(
+                #    main_device_str=str(ltx_main_device),
+                #    vae_device_str=str(ltx_vae_device)
+                #)
+                #logging.info(f"[SWAP] LTX service restored to Main: {ltx_main_device}, VAE: {ltx_vae_device}.")
 
 # --- PONTO DE ENTRADA E INSTANCIAÇÃO ---
 # A instância é criada na primeira importação.