Update deformes4D_engine.py

deformes4D_engine.py

@@ -169,9 +169,24 @@ class Deformes4DEngine:
             
             latent_fragments.append(latents_video)
         
-        logger.info("--- CONCATENANDO TODOS OS FRAGMENTOS LATENTES ---")
-        tensors_para_concatenar = [frag.to(self.device) for frag in latent_fragments]
+    
+        logger.info("--- CONCATENANDO nem TODOS OS FRAGMENTOS LATENTES ---")
+        tensors_para_concatenar = []
+        for idx, tensor_frag in enumerate(latent_fragments):
+            # Move cada tensor para o dispositivo de destino antes de adicioná-lo à lista.
+            target_device = self.device
+            tensor_on_target_device = tensor_frag.to(target_device)
+            if idx < len(latent_fragments) - 1:
+                tensors_para_concatenar.append(tensor_on_target_device[:, :, :-1, :, :]) 
+            else:
+                tensors_para_concatenar.append(tensor_on_target_device)
+
         processed_latents = torch.cat(tensors_para_concatenar, dim=2)
+        logger.info(f"Concatenação concluída. Shape final do tensor latente: {final_concatenated_latents.shape}")
+        
+        
+        
+        
         # [CORREÇÃO 2] Referência correta da variável no log
         logger.info(f"Concatenação concluída. Shape final do tensor latente: {processed_latents.shape}")