Update api/ltx_server.py

api/ltx_server.py

@@ -416,7 +416,8 @@ class VideoService:
         if num_latente_por_chunk >= sum_latent:
             return [latents_brutos.clone()]
     
-        steps = (sum_latent) // num_latente_por_chunk 
+        n_chunks = (sum_latent) // num_latente_por_chunk 
+        steps = sum_latent//n_chunks
         print("================PODA CAUSAL=================")
         print(f"[DEBUG] TOTAL LATENTES = {sum_latent}")
         print(f"[DEBUG] Num LATENTES por chunk = {num_latente_por_chunk}")
@@ -424,16 +425,16 @@ class VideoService:
 
         start = 0
         end = num_latente_por_chunk
-        i=1
+        
         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()
+            while i < steps:
+               start = i*num_latente_por_chunk
+               end = (start+num_latente_por_chunk)-1
+               if i+1 >= steps:
+                   end = sum_latent-1
+               chunk = latents_brutos[:, :, start:end, :, :].clone()
                chunks.append(chunk)
-               print(f"[DEBUG] chunk{i+1}[:, :, {start}:{end-1}, :, :] = {chunk.shape[2]}")
+               print(f"[DEBUG] chunk{i+1}[:, :, {start}:{end}, :, :] = {chunk.shape[2]}")
                start += steps
                i += 1
         else: