Update deformes4D_engine.py
Browse files- deformes4D_engine.py +1 -1
deformes4D_engine.py
CHANGED
|
@@ -238,7 +238,7 @@ class Deformes4DEngine:
|
|
| 238 |
target_device = self.device
|
| 239 |
for idx, tensor_frag in enumerate(list_latents_fragments):
|
| 240 |
tensor_on_target_device = tensor_frag.to(target_device)
|
| 241 |
-
if idx < len(
|
| 242 |
tensors_para_concatenar.append(tensor_on_target_device[:, :, :-1, :, :])
|
| 243 |
else:
|
| 244 |
tensors_para_concatenar.append(tensor_on_target_device)
|
|
|
|
| 238 |
target_device = self.device
|
| 239 |
for idx, tensor_frag in enumerate(list_latents_fragments):
|
| 240 |
tensor_on_target_device = tensor_frag.to(target_device)
|
| 241 |
+
if idx < len(upscaled_latent_fragments) - 1:
|
| 242 |
tensors_para_concatenar.append(tensor_on_target_device[:, :, :-1, :, :])
|
| 243 |
else:
|
| 244 |
tensors_para_concatenar.append(tensor_on_target_device)
|