Update api/ltx_server.py

api/ltx_server.py

@@ -414,7 +414,7 @@ class VideoService:
         chunks = []
     
         if num_latente_por_chunk >= sum_latent:
-            return [latents_brutos.clone()]
+            return [latents_brutos]
     
         n_chunks = (sum_latent) // num_latente_por_chunk 
         steps = sum_latent//n_chunks
@@ -429,13 +429,13 @@ class VideoService:
             end=1
             while i < n_chunks:
                 start += end
-                end = num_latente_por_chunk
-                if start+end+3>=sum_latent-1:
-                    end = (sum_latent-1)-start
+                end = start+num_latente_por_chunk
+                if end+3>=sum_latent-1:
+                    end = (sum_latent-1)
                     i = n_chunks
                 else:
                     i += 1
-                chunk = latents_brutos[:, :, start-1:end, :, :].clone()
+                chunk = latents_brutos[:, :, start-1:end, :, :].clone().detach())
                 chunks.append(chunk)
                 print(f"[DEBUG] chunk{i}[:, :, {start-1}:{end}, :, :] = {chunk.shape[2]}")
         else: