Update api/ltx_server.py

api/ltx_server.py

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