Update deformes4D_engine.py
Browse files- 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:
|