Update api/ltx_server.py

api/ltx_server.py

@@ -426,15 +426,18 @@ class VideoService:
 
         if steps > 1:
             i = 0
+            end=1
             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}, :, :] = {chunk.shape[2]}")
-               i += 1
+                start = end-1
+                end += num_latente_por_chunk
+                if end+2>=sum_latent:
+                    end = sum_latent-1
+                    i = steps
+                else:
+                    i += 1
+                chunk = latents_brutos[:, :, start:end, :, :].clone()
+                chunks.append(chunk)
+                print(f"[DEBUG] chunk{i}[:, :, {start}:{end}, :, :] = {chunk.shape[2]}")
         else:
             print(f"[DEBUG] numero chunks minimo")
             print(f"[DEBUG] latents_brutos[:, :, :, :, :] = {latents_brutos.shape[2]}")