Spaces:
Runtime error
Runtime error
1 frame
Browse files- 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 |
-
|
| 594 |
#generated_latents = torch.cat([video_latents[:, :, -1:].to(generated_latents), generated_latents], dim=2)
|
| 595 |
-
|
| 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)
|