EuuIia commited on
Commit
64bb5d1
·
verified ·
1 Parent(s): 4f6d48f

Update api/ltx_server.py

Browse files
Files changed (1) hide show
  1. api/ltx_server.py +3 -3
api/ltx_server.py CHANGED
@@ -420,16 +420,16 @@ class VideoService:
420
  print(f"[DEBUG] TOTAL LATENTES A DIVIDIR /{quantidade} = {sum_latent}")
421
 
422
  for i in range(quantidade):
423
- start = i * step
424
  end = start + step
425
  if i == quantidade - 1:
426
- end = total
427
  else:
428
  end += overlap
429
  chunk = latents_brutos[:, :, start:end, :, :].clone()
430
  chunks.append(chunk)
431
  chunk_sum = chunk.shape[2]
432
- print(f"[DEBUG] chunk{i+1}[:, :, start:end, :, :] = {chunk_sum}")
433
 
434
  print(f"================PODA CAUSAL=================")
435
  return chunks
 
420
  print(f"[DEBUG] TOTAL LATENTES A DIVIDIR /{quantidade} = {sum_latent}")
421
 
422
  for i in range(quantidade):
423
+ start = i * step
424
  end = start + step
425
  if i == quantidade - 1:
426
+ end = sum_latent
427
  else:
428
  end += overlap
429
  chunk = latents_brutos[:, :, start:end, :, :].clone()
430
  chunks.append(chunk)
431
  chunk_sum = chunk.shape[2]
432
+ print(f"[DEBUG] chunk{i+1}[:, :, {start}:{end}, :, :] = {chunk_sum}")
433
 
434
  print(f"================PODA CAUSAL=================")
435
  return chunks