euiia commited on
Commit
9656f0a
·
verified ·
1 Parent(s): 9af8c0d

Update deformes4D_engine.py

Browse files
Files changed (1) hide show
  1. deformes4D_engine.py +3 -1
deformes4D_engine.py CHANGED
@@ -262,8 +262,10 @@ class Deformes4DEngine:
262
  temp_video_base_name = f"debug_chunk_{chunk_idx}"
263
 
264
  # Converte o latente do chunk em um vídeo MP4
265
- temp_video_path = self._generate_video_from_latents(single_chunk_latent, temp_video_base_name)
266
 
 
 
267
  # Conta os frames no vídeo gerado
268
  if os.path.exists(temp_video_path):
269
  with imageio.get_reader(temp_video_path) as reader:
 
262
  temp_video_base_name = f"debug_chunk_{chunk_idx}"
263
 
264
  # Converte o latente do chunk em um vídeo MP4
265
+ pixel_tensor = self.latents_to_pixels(single_chunk_latent)
266
 
267
+ temp_video_path = self.save_video_from_tensor(pixel_tensor, temp_video_base_name, fps=24)
268
+
269
  # Conta os frames no vídeo gerado
270
  if os.path.exists(temp_video_path):
271
  with imageio.get_reader(temp_video_path) as reader: