euiia commited on
Commit
351cd3f
·
verified ·
1 Parent(s): 3c20c7f

Update deformes4D_engine.py

Browse files
Files changed (1) hide show
  1. deformes4D_engine.py +3 -0
deformes4D_engine.py CHANGED
@@ -375,6 +375,8 @@ class Deformes4DEngine:
375
  #self.concatenate_videos_ffmpeg(video_clips_paths, final_movie_path)
376
 
377
 
 
 
378
  # 4. Concatenação e Renderização Final (A Lógica do Teste)
379
  logger.info("--- CONCATENANDO TODOS OS FRAGMENTOS LATENTES ---")
380
 
@@ -382,6 +384,7 @@ class Deformes4DEngine:
382
  for idx, tensor_frag in enumerate(low_res_latent_fragments):
383
  # --- A CORREÇÃO ESTÁ AQUI ---
384
  # Move cada tensor para o dispositivo de destino antes de adicioná-lo à lista.
 
385
  tensor_on_target_device = tensor_frag.to(target_device)
386
 
387
  if idx < len(low_res_latent_fragments) - 1:
 
375
  #self.concatenate_videos_ffmpeg(video_clips_paths, final_movie_path)
376
 
377
 
378
+
379
+
380
  # 4. Concatenação e Renderização Final (A Lógica do Teste)
381
  logger.info("--- CONCATENANDO TODOS OS FRAGMENTOS LATENTES ---")
382
 
 
384
  for idx, tensor_frag in enumerate(low_res_latent_fragments):
385
  # --- A CORREÇÃO ESTÁ AQUI ---
386
  # Move cada tensor para o dispositivo de destino antes de adicioná-lo à lista.
387
+ target_device = self.device
388
  tensor_on_target_device = tensor_frag.to(target_device)
389
 
390
  if idx < len(low_res_latent_fragments) - 1: