Fabrice-TIERCELIN commited on
Commit
3847094
·
verified ·
1 Parent(s): fcec7ca
Files changed (1) hide show
  1. app_endframe.py +2 -2
app_endframe.py CHANGED
@@ -590,9 +590,9 @@ def worker(input_video, end_frame, end_frame_weight, prompt, n_prompt, seed, bat
590
  callback=callback,
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[:, :, 2:]
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)
 
590
  callback=callback,
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)