Update api/ltx_server.py

api/ltx_server.py

@@ -884,7 +884,8 @@ class VideoService:
                         # # --- ETAPA 3: REFINAMENTO DE TEXTURA (SECOND PASS) ---
                         print(f"\n--- INICIANDO ETAPA 3/{cc} ")
 
-                        second_pass_kwargs = first_pass_config.copy()
+                        first_pass_kwargs = call_kwargs.copy()
+    
                         second_pass_config = self.config.get("second_pass", {}).copy()
 
                         second_pass_width = downscaled_width * 2
@@ -931,10 +932,10 @@ class VideoService:
                     single_pass_kwargs["output_type"] = "latent"
                     
                     # Remove keys that might conflict or are not used in single pass / handled by above
-                    single_pass_kwargs.pop("num_inference_steps", None) 
-                    single_pass_kwargs.pop("first_pass", None) 
-                    single_pass_kwargs.pop("second_pass", None)
-                    single_pass_kwargs.pop("downscale_factor", None)
+                    #single_pass_kwargs.pop("num_inference_steps", None) 
+                    #single_pass_kwargs.pop("first_pass", None) 
+                    #single_pass_kwargs.pop("second_pass", None)
+                    #single_pass_kwargs.pop("downscale_factor", None)
                     
                     latents = self.pipeline(**single_pass_kwargs).images
                     log_tensor_info(latents, "Latentes Finais (Etapa Única)")