Update api/ltx_server.py

api/ltx_server.py

@@ -571,7 +571,7 @@ class VideoService:
     
         n_chunks = (sum_latent) // num_latente_por_chunk 
         steps = sum_latent//n_chunks
-        print("================PODA CAUSAL=================")
+        print("==========PODA CAUSAL[start:stop-1]==========")
         print(f"[DEBUG] TOTAL LATENTES = {sum_latent}")
         print(f"[DEBUG] LATENTES min por chunk = {num_latente_por_chunk}")
         print(f"[DEBUG] Número de chunks = {n_chunks}")
@@ -581,18 +581,22 @@ class VideoService:
                 start = (num_latente_por_chunk*i)
                 end = (start+num_latente_por_chunk+overlap)
                 if i+1 < n_chunks:
-                    chunk = latents_brutos[:, :, start:end, :, :].clone().detach()
-                    print(f"[DEBUG] chunk{i+1}[:, :, {start}:{end}, :, :] = {chunk.shape[2]}")
+                    if i > 0:
+                        chunk = latents_brutos[:, :, start-overlap:end, :, :].clone().detach()
+                        print(f"[DEBUG] chunk{i+1}[:, :, {start}:{end}, :, :] = {chunk.shape[2]}")
+                    else:
+                        chunk = latents_brutos[:, :, start:end, :, :].clone().detach()
+                        print(f"[DEBUG] chunk{i+1}[:, :, {start}:{end}, :, :] = {chunk.shape[2]}")
                 else:
-                    chunk = latents_brutos[:, :, start:, :, :].clone().detach()
-                    print(f"[DEBUG] chunk{i+1}[:, :, {start}:, :, :] = {chunk.shape[2]}")
+                    chunk = latents_brutos[:, :, start-overlap:, :, :].clone().detach()
+                    print(f"[DEBUG] chunk{i+1}[:, :, {start-overlap}:, :, :] = {chunk.shape[2]}")
                 chunks.append(chunk)
                 i+=1
         else:
             print(f"[DEBUG] numero chunks minimo ")
             print(f"[DEBUG] latents_brutos[:, :, :, :, :] = {latents_brutos.shape[2]}")
             chunks.append(latents_brutos)
-        print("================PODA CAUSAL=================")
+        print("\n\n================PODA CAUSAL=================")
         return chunks
    
     def _get_total_frames(self, video_path: str) -> int:
@@ -621,10 +625,16 @@ class VideoService:
         video_fade_ini = None
         nova_lista = []
 
-        print("===========CONCATECAO CAUSAL=============")
+        if crossfade_frames == 0:
+            print("\n\n[DEBUG] CROSSFADE_FRAMES=0 Ship concatenation causal")
+            return video_paths
+        
+
+        print("\n\n===========CONCATECAO CAUSAL=============")
      
         print(f"[DEBUG] Iniciando pipeline com {total_partes} vídeos e {poda} frames de crossfade")
-    
+
+        
         for i in range(total_partes):
             base = video_paths[i]
         
@@ -683,7 +693,7 @@ class VideoService:
            
             nova_lista.append(video_podado)
             print(f"[DEBUG] Video podado {i+1} adicionado {self._get_total_frames(video_podado)} frames ✅")
-               
+
 
 
         print("===========CONCATECAO CAUSAL=============")
@@ -947,7 +957,7 @@ class VideoService:
                 except Exception:
                     pass
                     
-                latents_parts_vae = self._dividir_latentes_por_tamanho(latents_cpu_vae,15,1)
+                latents_parts_vae = self._dividir_latentes_por_tamanho(latents_cpu_vae,4,1)
                 
                 for latents in latents_parts_vae:
                     print(f"[DEBUG] Partição {par}: {tuple(latents.shape)}")