euiia commited on
Commit
93a4143
·
verified ·
1 Parent(s): 33e3430

Update deformes4D_engine.py

Browse files
Files changed (1) hide show
  1. 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(low_res_latent_fragments) - 1:
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)