Update api/ltx_server.py

api/ltx_server.py

@@ -416,25 +416,28 @@ class VideoService:
         if num_latente_por_chunk >= sum_latent:
             return [latents_brutos.clone()]
     
-        steps = (sum_latent -2 ) // num_latente_por_chunk 
+        steps = (sum_latent) // num_latente_por_chunk 
         print("================PODA CAUSAL=================")
         print(f"[DEBUG] TOTAL LATENTES = {sum_latent}")
         print(f"[DEBUG] Num LATENTES por chunk = {num_latente_por_chunk}")
         print(f"[DEBUG] Número de chunks = {steps}")
-    
-        for i in range(steps):
-            start = i * num_latente_por_chunk
-            end = start + num_latente_por_chunk
-            #if i > 0:
-            #    start = overlap  # sobreposição
-            if end >= sum_latent:
-                end = sum_latent
-            else:
-                end = overlap
-            chunk = latents_brutos[:, :, start:end, :, :].clone()
-            chunks.append(chunk)
-            print(f"[DEBUG] chunk{i+1}[:, :, {start}:{end}, :, :] = {chunk.shape[2]}")
-    
+
+        start = 0
+        end = num_latente_por_chunk
+        if steps > 1:
+            while start+steps < sum_latent:
+               end = start+steps
+               if end+3>sum_latent:
+                  end = sum_latent
+                  start = end
+               chunk = latents_brutos[:, :, start:end-1, :, :].clone()
+               chunks.append(chunk)
+               print(f"[DEBUG] chunk{i+1}[:, :, {start}:{end-1}, :, :] = {chunk.shape[2]}")
+               start += steps
+        else:
+            print(f"[DEBUG] numero chunks minimo")
+            print(f"[DEBUG] latents_brutos[:, :, :, :, :] = {latents_brutos.shape[2]}")
+            chunks.append(latents_brutos)
         print("================PODA CAUSAL=================")
         return chunks