Update deformes4D_engine.py
Browse files- 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(
|
| 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[
|
| 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 |
}
|