Update api/ltx_server.py

api/ltx_server.py

@@ -599,21 +599,28 @@ class VideoService:
                 #    }
                 #)
 
+                # 
                 first_pass_kwargs = call_kwargs.copy()
                 first_pass_kwargs.update(first_pass_args)
                 
+                print("[DEBUG] Executando FIRST PASS (pipeline base)...")
                 with ctx:
-                    _apply_precision_policyresult_first = self.pipeline(**first_pass_kwargs)
-                
-                latents_first = result_first.latents
-
+                     result_first = self.pipeline(**first_pass_kwargs)
+            
+                latents_first = result_first.latents if hasattr(result_first, "latents") else result_first
+                print(f"[DEBUG] Latentes FIRST PASS: {tuple(latents_first.shape)}")
+            
+                # --- SECOND PASS ---
+                print("[DEBUG] Executando SECOND PASS (latent_upsampler)...")
                 with ctx:
-                    result_second = self.latent_upsampler(
+                        result_second = self.latent_upsampler(
                         latents=latents_first,
                         **second_pass_args
                     )
+            
                 latents_final = result_second.latents if hasattr(result_second, "latents") else result_second
-
+                print(f"[DEBUG] Latentes SECOND PASS: {tuple(latents_final.shape)}")
+            
                 
                 print("[DEBUG] Chamando multi_scale_pipeline...")
                 t_ms = time.perf_counter()