Eueuiaa commited on
Commit
d448752
·
verified ·
1 Parent(s): 43129bd

Update api/ltx_server.py

Browse files
Files changed (1) hide show
  1. api/ltx_server.py +6 -6
api/ltx_server.py CHANGED
@@ -331,7 +331,7 @@ class VideoService:
331
  self.pipeline, self.latent_upsampler = self._load_models()
332
  print(f"[DEBUG] Pipeline e Upsampler carregados. Upsampler ativo? {bool(self.latent_upsampler)}")
333
 
334
- self._apply_ltx_pipeline_patches()
335
 
336
  print(f"[DEBUG] Movendo modelos para {self.device}...")
337
  self.pipeline.to(self.device)
@@ -949,13 +949,13 @@ class VideoService:
949
  #latents_parts_up = self._dividir_latentes_por_tamanho(latents_cpu_up,15,0)
950
 
951
 
952
- #lat_aup, lat_bup = self._dividir_latentes(latents_cpu_up)
953
- #print(f"[DEBUG] Partição Aup: {tuple(lat_aup.shape)}")
954
- #print(f"[DEBUG] Partição Bup: {tuple(lat_bup.shape)}")
955
 
956
- #latents_parts_up = [lat_aup, lat_bup]
957
 
958
- latents_parts_up = [latents_cpu_up]
959
 
960
 
961
  for latents in latents_parts_up:
 
331
  self.pipeline, self.latent_upsampler = self._load_models()
332
  print(f"[DEBUG] Pipeline e Upsampler carregados. Upsampler ativo? {bool(self.latent_upsampler)}")
333
 
334
+ #self._apply_ltx_pipeline_patches()
335
 
336
  print(f"[DEBUG] Movendo modelos para {self.device}...")
337
  self.pipeline.to(self.device)
 
949
  #latents_parts_up = self._dividir_latentes_por_tamanho(latents_cpu_up,15,0)
950
 
951
 
952
+ lat_aup, lat_bup = self._dividir_latentes(latents_cpu_up)
953
+ print(f"[DEBUG] Partição Aup: {tuple(lat_aup.shape)}")
954
+ print(f"[DEBUG] Partição Bup: {tuple(lat_bup.shape)}")
955
 
956
+ latents_parts_up = [lat_aup, lat_bup]
957
 
958
+ #latents_parts_up = [latents_cpu_up]
959
 
960
 
961
  for latents in latents_parts_up: