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