Carlos s commited on
Commit
79febce
·
verified ·
1 Parent(s): 85f06ad

Update api/ltx_server.py

Browse files
Files changed (1) hide show
  1. api/ltx_server.py +13 -14
api/ltx_server.py CHANGED
@@ -428,24 +428,23 @@ class VideoService:
428
  partes.append(parte)
429
 
430
  return partes
431
-
432
 
433
- def _dividir_latentes(self, latents_brutos):
434
- total = latents_brutos.shape[2] # dimensão temporal (número de latentes)
435
-
 
436
  if total % 2 == 1: # ÍMPAR
437
- # Ex: 11 primeira 0..5, segunda 5..10
438
- cut = total // 2
439
- primeira = latents_brutos[:, :, :cut+1, :, :].clone()
440
- segunda = latents_brutos[:, :, cut:, :, :].clone()
441
-
442
  else: # PAR
443
- # Ex: 12 primeira 0..5, segunda 5..11
444
- cut = total // 2
445
- primeira = latents_brutos[:, :, :cut, :, :].clone()
446
- segunda = latents_brutos[:, :, cut-1:, :, :].clone()
447
-
448
  return primeira, segunda
 
449
 
450
  def _concat_mp4s_no_reencode(self, mp4_list: List[str], out_path: str):
451
  """
 
428
  partes.append(parte)
429
 
430
  return partes
 
431
 
432
+
433
+ def dividir_latentes(latents_brutos):
434
+ total = latents_brutos.shape[2] # dimensão temporal (latentes)
435
+
436
  if total % 2 == 1: # ÍMPAR
437
+ cut = total // 2
438
+ primeira = latents_brutos[:, :, :cut+1, :, :].clone()
439
+ segunda = latents_brutos[:, :, cut:, :, :].clone()
 
 
440
  else: # PAR
441
+ cut = total // 2
442
+ # primeira parte até o meio, mas o último frame deve ser ajustado
443
+ primeira = latents_brutos[:, :, :cut+1, :, :].clone()
444
+ segunda = latents_brutos[:, :, cut:, :, :].clone()
445
+
446
  return primeira, segunda
447
+
448
 
449
  def _concat_mp4s_no_reencode(self, mp4_list: List[str], out_path: str):
450
  """