Carlos s commited on
Commit
248af78
·
verified ·
1 Parent(s): c77141b

Update api/ltx_server.py

Browse files
Files changed (1) hide show
  1. api/ltx_server.py +5 -5
api/ltx_server.py CHANGED
@@ -429,15 +429,15 @@ class VideoService:
429
  end=1
430
  while i < n_chunks:
431
  start = end
432
- end += num_latente_por_chunk
433
- if end+2>=sum_latent:
434
- end = sum_latent-1
435
  i = n_chunks
436
  else:
437
  i += 1
438
- chunk = latents_brutos[:, :, start-1:end+1, :, :].clone()
439
  chunks.append(chunk)
440
- print(f"[DEBUG] chunk{i}[:, :, {start-1}:{end+1}, :, :] = {chunk.shape[2]}")
441
  else:
442
  print(f"[DEBUG] numero chunks minimo")
443
  print(f"[DEBUG] latents_brutos[:, :, :, :, :] = {latents_brutos.shape[2]}")
 
429
  end=1
430
  while i < n_chunks:
431
  start = end
432
+ end = num_latente_por_chunk
433
+ if start+end+3>=sum_latent-1:
434
+ end = (sum_latent-1)-start
435
  i = n_chunks
436
  else:
437
  i += 1
438
+ chunk = latents_brutos[:, :, start-1:end, :, :].clone()
439
  chunks.append(chunk)
440
+ print(f"[DEBUG] chunk{i}[:, :, {start-1}:{end}, :, :] = {chunk.shape[2]}")
441
  else:
442
  print(f"[DEBUG] numero chunks minimo")
443
  print(f"[DEBUG] latents_brutos[:, :, :, :, :] = {latents_brutos.shape[2]}")