Eueuiaa commited on
Commit
3bb53ba
·
verified ·
1 Parent(s): 0bde19b

Update api/ltx_server.py

Browse files
Files changed (1) hide show
  1. api/ltx_server.py +8 -4
api/ltx_server.py CHANGED
@@ -578,14 +578,18 @@ class VideoService:
578
  if n_chunks > 1:
579
  i=0
580
  while i < n_chunks:
 
 
 
 
581
  start = (num_latente_por_chunk*i)
582
  end = (start+num_latente_por_chunk+(overlap+1))
583
  if i+1 < n_chunks:
584
- chunk = latents_brutos[:, :, start:end, :, :].clone().detach()
585
- print(f"[DEBUG] chunk{i+1}[:, :, {start}:{end}, :, :] = {chunk.shape[2]}")
586
  else:
587
- chunk = latents_brutos[:, :, start:, :, :].clone().detach()
588
- print(f"[DEBUG] chunk{i+1}[:, :, {start}:, :, :] = {chunk.shape[2]}")
589
  chunks.append(chunk)
590
  i+=1
591
  else:
 
578
  if n_chunks > 1:
579
  i=0
580
  while i < n_chunks:
581
+ if i>0:
582
+ dow=1
583
+ else:
584
+ dow=0
585
  start = (num_latente_por_chunk*i)
586
  end = (start+num_latente_por_chunk+(overlap+1))
587
  if i+1 < n_chunks:
588
+ chunk = latents_brutos[:, :, start-(dow):end, :, :].clone().detach()
589
+ print(f"[DEBUG] chunk{i+1}[:, :, {start-dow}:{end}, :, :] = {chunk.shape[2]}")
590
  else:
591
+ chunk = latents_brutos[:, :, start-(dow):, :, :].clone().detach()
592
+ print(f"[DEBUG] chunk{i+1}[:, :, {start-(dow)}:, :, :] = {chunk.shape[2]}")
593
  chunks.append(chunk)
594
  i+=1
595
  else: