Fabrice-TIERCELIN commited on
Commit
1b40ecf
·
verified ·
1 Parent(s): 3847094
Files changed (1) hide show
  1. app_endframe.py +1 -2
app_endframe.py CHANGED
@@ -591,8 +591,7 @@ def worker(input_video, end_frame, end_frame_weight, prompt, n_prompt, seed, bat
591
  )
592
 
593
  if is_start_of_video:
594
- #generated_latents = torch.cat([video_latents[:, :, -1:].to(generated_latents), generated_latents], dim=2)
595
- generated_latents = generated_latents[:, :, 1:]
596
 
597
  total_generated_latent_frames += int(generated_latents.shape[2])
598
  history_latents = torch.cat([generated_latents.to(history_latents), history_latents], dim=2)
 
591
  )
592
 
593
  if is_start_of_video:
594
+ generated_latents = torch.cat([video_latents[:, :, -1:].to(generated_latents), generated_latents], dim=2)
 
595
 
596
  total_generated_latent_frames += int(generated_latents.shape[2])
597
  history_latents = torch.cat([generated_latents.to(history_latents), history_latents], dim=2)