Fabrice-TIERCELIN commited on
Commit
8cb4781
·
verified ·
1 Parent(s): b910ee3

End latent

Browse files
Files changed (1) hide show
  1. app.py +2 -2
app.py CHANGED
@@ -1358,8 +1358,8 @@ def worker_video_end(input_video, end_frame, prompts, n_prompt, seed, batch, res
1358
  if effective_clean_frames > 0 and split_idx < len(splits):
1359
  clean_latents_1x = splits[split_idx]
1360
 
1361
- if end_frame is not None:
1362
- clean_latents = torch.cat([start_latent, clean_latents_1x, end_frame], dim=2)
1363
  else:
1364
  clean_latents = torch.cat([start_latent, clean_latents_1x], dim=2)
1365
 
 
1358
  if effective_clean_frames > 0 and split_idx < len(splits):
1359
  clean_latents_1x = splits[split_idx]
1360
 
1361
+ if end_latent is not None:
1362
+ clean_latents = torch.cat([start_latent, clean_latents_1x, end_latent], dim=2)
1363
  else:
1364
  clean_latents = torch.cat([start_latent, clean_latents_1x], dim=2)
1365