Aduc_sdr

Paused

EU-IA commited on Aug 29

Commit

5b8b9c0

verified ·

1 Parent(s): 85ed61a

Update deformes4D_engine.py

Files changed (1) hide show

deformes4D_engine.py CHANGED Viewed

@@ -256,14 +256,14 @@ class Deformes4DEngine:
             if is_first_fragment:
                 if n_trim_latents > 0 and latents_for_video.shape[2] > n_trim_latents:
-                     latents_for_video = latents_for_video[:, :, :-n_trim_latents, :, :]  # Nota: Cortar 24 do fim
             elif is_penultimate_fragment:
                 if echo_frames > 0 and latents_for_video.shape[2] > echo_frames:
                      latents_for_video = latents_for_video[:, :, echo_frames:, :, :]  # Cortar eco do início
             if n_trim_latents > 0 and latents_for_video.shape[2] > n_trim_latents:
-                latents_for_video = latents_for_video[:, :, :-n_trim_latents, :, :]  # Cortar 24 do fim
             else:
                 if echo_frames > 0 and latents_for_video.shape[2] > echo_frames:

             if is_first_fragment:
                 if n_trim_latents > 0 and latents_for_video.shape[2] > n_trim_latents:
+                     latents_for_video = latents_for_video[:, :, :n_trim_latents, :, :]  # Nota: Cortar 24 do fim
             elif is_penultimate_fragment:
                 if echo_frames > 0 and latents_for_video.shape[2] > echo_frames:
                      latents_for_video = latents_for_video[:, :, echo_frames:, :, :]  # Cortar eco do início
             if n_trim_latents > 0 and latents_for_video.shape[2] > n_trim_latents:
+                latents_for_video = latents_for_video[:, :, :n_trim_latents, :, :]  # Cortar 24 do fim
             else:
                 if echo_frames > 0 and latents_for_video.shape[2] > echo_frames: