Spaces:
Running
Running
Original
Browse files- 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 |
-
|
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)
|