Fabrice-TIERCELIN commited on
Commit
7e7d552
·
verified ·
1 Parent(s): 46b67f9
Files changed (1) hide show
  1. app.py +1 -2
app.py CHANGED
@@ -428,12 +428,11 @@ def worker(input_image, image_position, prompts, n_prompt, seed, resolution, tot
428
  def callback(d):
429
  return
430
 
 
431
  if is_last_frame:
432
- indices = torch.arange(sum([1, 16, 2, 1, latent_window_size]) - 1, -1, -1).unsqueeze(0)
433
  latent_indices, clean_latent_1x_indices, clean_latent_2x_indices, clean_latent_4x_indices, clean_latent_indices_start = indices.split([latent_window_size, 1, 2, 16, 1], dim=1)
434
  clean_latent_indices = torch.cat([clean_latent_1x_indices, clean_latent_indices_start], dim=1)
435
  else:
436
- indices = torch.arange(0, sum([1, 16, 2, 1, latent_window_size])).unsqueeze(0)
437
  clean_latent_indices_start, clean_latent_4x_indices, clean_latent_2x_indices, clean_latent_1x_indices, latent_indices = indices.split([1, 16, 2, 1, latent_window_size], dim=1)
438
  clean_latent_indices = torch.cat([clean_latent_indices_start, clean_latent_1x_indices], dim=1)
439
 
 
428
  def callback(d):
429
  return
430
 
431
+ indices = torch.arange(0, sum([1, 16, 2, 1, latent_window_size])).unsqueeze(0)
432
  if is_last_frame:
 
433
  latent_indices, clean_latent_1x_indices, clean_latent_2x_indices, clean_latent_4x_indices, clean_latent_indices_start = indices.split([latent_window_size, 1, 2, 16, 1], dim=1)
434
  clean_latent_indices = torch.cat([clean_latent_1x_indices, clean_latent_indices_start], dim=1)
435
  else:
 
436
  clean_latent_indices_start, clean_latent_4x_indices, clean_latent_2x_indices, clean_latent_1x_indices, latent_indices = indices.split([1, 16, 2, 1, latent_window_size], dim=1)
437
  clean_latent_indices = torch.cat([clean_latent_indices_start, clean_latent_1x_indices], dim=1)
438