Update api/ltx_server.py

api/ltx_server.py

@@ -424,15 +424,15 @@ class VideoService:
         print(f"[DEBUG] Número de chunks = {n_chunks}")
         if n_chunks > 1:
             i=0
-            while i < n_chunks-1:
-                start = num_latente_por_chunk*i
-                end = start+num_latente_por_chunk+overlap
-                if end < sum_latent-3:
+            while i < n_chunks:
+                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}[:, :, {start}:{end}, :, :] = {chunk.shape[2]}")
+                    print(f"[DEBUG] chunk{i+1}[:, :, {start}:{end}, :, :] = {chunk.shape[2]}")
                 else:
                     chunk = latents_brutos[:, :, start:, :, :].clone().detach()
-                    print(f"[DEBUG] chunk{i}[:, :, {start}:, :, :] = {chunk.shape[2]}")
+                    print(f"[DEBUG] chunk{i+1}[:, :, {start}:, :, :] = {chunk.shape[2]}")
                 chunks.append(chunk)
                 i+=1
         else: