euiia commited on
Commit
63ceaa5
·
verified ·
1 Parent(s): 5137a03

Update deformes4D_engine.py

Browse files
Files changed (1) hide show
  1. deformes4D_engine.py +2 -2
deformes4D_engine.py CHANGED
@@ -236,7 +236,7 @@ class Deformes4DEngine:
236
  progress((num_transitions_to_generate) / (num_transitions_to_generate + 2), desc="Concatenando latentes...")
237
  tensors_para_concatenar = []
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, :, :])
@@ -327,7 +327,7 @@ class Deformes4DEngine:
327
 
328
  def _generate_latent_tensor_internal(self, conditioning_items, ltx_params, target_resolution, total_frames_to_generate):
329
  kwargs = {
330
- **ltx_params, 'width': target_resolution[0], 'height': target_resolution[1],
331
  'video_total_frames': total_frames_to_generate, 'video_fps': 24,
332
  'current_fragment_index': int(time.time()), 'conditioning_items_data': conditioning_items
333
  }
 
236
  progress((num_transitions_to_generate) / (num_transitions_to_generate + 2), desc="Concatenando latentes...")
237
  tensors_para_concatenar = []
238
  target_device = self.device
239
+ for idx, tensor_frag in enumerate(upscaled_latent_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, :, :])
 
327
 
328
  def _generate_latent_tensor_internal(self, conditioning_items, ltx_params, target_resolution, total_frames_to_generate):
329
  kwargs = {
330
+ **ltx_params, 'width': target_resolution[1], 'height': target_resolution[0],
331
  'video_total_frames': total_frames_to_generate, 'video_fps': 24,
332
  'current_fragment_index': int(time.time()), 'conditioning_items_data': conditioning_items
333
  }