eeuuia commited on
Commit
a74cf78
·
verified ·
1 Parent(s): fe09ee4

Update api/ltx/ltx_aduc_pipeline.py

Browse files
Files changed (1) hide show
  1. api/ltx/ltx_aduc_pipeline.py +8 -4
api/ltx/ltx_aduc_pipeline.py CHANGED
@@ -199,6 +199,7 @@ class LtxAducPipeline:
199
  # --- ETAPA 1: GERAÇÃO DE CHUNKS E SALVAMENTO ---
200
  for i, chunk_prompt in enumerate(prompt_list):
201
  logging.info(f"Processing scene {i+1}/{num_chunks}: '{chunk_prompt[:50]}...'")
 
202
  call_kwargs["prompt"] = chunk_prompt
203
 
204
  with torch.autocast(device_type=self.main_device.type, dtype=self.runtime_autocast_dtype, enabled="cuda" in self.main_device.type):
@@ -207,11 +208,14 @@ class LtxAducPipeline:
207
 
208
  if is_narrative and i < num_chunks - 1:
209
  overlap_latents = chunk_latents[:, :, -overlap_frames:, :, :].clone()
210
- call_kwargs["conditioning_items"] = [LatentConditioningItem(overlap_latents, 0, 1.0)]
211
  else:
212
- call_kwargs.pop("conditioning_items", None)
213
-
214
- if i > 0: chunk_latents = chunk_latents[:, :, overlap_frames:, :, :]
 
 
 
215
 
216
  chunk_path = RESULTS_DIR / f"temp_chunk_{i}_{used_seed}.pt"
217
  # --- NOVO: Rastreia o arquivo para limpeza ---
 
199
  # --- ETAPA 1: GERAÇÃO DE CHUNKS E SALVAMENTO ---
200
  for i, chunk_prompt in enumerate(prompt_list):
201
  logging.info(f"Processing scene {i+1}/{num_chunks}: '{chunk_prompt[:50]}...'")
202
+ call_kwargs.pop("prompt", None)
203
  call_kwargs["prompt"] = chunk_prompt
204
 
205
  with torch.autocast(device_type=self.main_device.type, dtype=self.runtime_autocast_dtype, enabled="cuda" in self.main_device.type):
 
208
 
209
  if is_narrative and i < num_chunks - 1:
210
  overlap_latents = chunk_latents[:, :, -overlap_frames:, :, :].clone()
211
+ #call_kwargs["conditioning_items"] = [LatentConditioningItem(overlap_latents, 0, 1.0)]
212
  else:
213
+ #call_kwargs.pop("conditioning_items", None)
214
+
215
+ print (f"overlap_latents {overlap_latents.shape}")
216
+ print (f"chunk_latents {chunk_latents.shape}")
217
+
218
+ #if i > 0: chunk_latents = chunk_latents[:, :, overlap_frames:, :, :]
219
 
220
  chunk_path = RESULTS_DIR / f"temp_chunk_{i}_{used_seed}.pt"
221
  # --- NOVO: Rastreia o arquivo para limpeza ---