Update api/ltx_server.py

api/ltx_server.py

@@ -420,9 +420,8 @@ class VideoService:
         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}")
+        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
@@ -430,16 +429,16 @@ class VideoService:
             while i < n_chunks:
                 start += end
                 end = start+num_latente_por_chunk
-                if end+3>=sum_latent-1:
-                    end = ""
-                    i = n_chunks
+                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]}")
                 else:
-                    i += 1
-                chunk = latents_brutos[:, :, start-1:end+1, :, :].clone().detach()
+                    i = n_chunks
+                    chunk = latents_brutos[:, :, start-1:, :, :].clone().detach()
+                    print(f"[DEBUG] chunk{i}[:, :, {start-1}:, :, :] = {chunk.shape[2]}")
                 chunks.append(chunk)
-                print(f"[DEBUG] chunk{i}[:, :, {start-1}:{end+1}, :, :] = {chunk.shape[2]}")
         else:
-            print(f"[DEBUG] numero chunks minimo")
+            print(f"[DEBUG] numero chunks minimo ")
             print(f"[DEBUG] latents_brutos[:, :, :, :, :] = {latents_brutos.shape[2]}")
             chunks.append(latents_brutos)
         print("================PODA CAUSAL=================")