Update app.py

app.py

@@ -42,18 +42,16 @@ prepare_latents_original = pipe.prepare_latents
 
 # pipe.prepare_latents = prepare_latents_loop
 
-# with the shift it will become step=0 0,1,2,3 -> step=1 1,2,3,0 -> step=2 2,3,0,1 -> step=3 3,0,1,2 -> step=4 0,1,2,3
-# so we only shift (N_FRAME-1)//8+1 times
-
 def modify_latents_callback(pipeline, step, timestep, callback_kwargs):
     print("Rolling latents on step", step)
     latents = callback_kwargs.get("latents")
     unpacked_latents = pipeline._unpack_latents(latents, (N_FRAME-1)//8+1, HEIGHT//32, WIDTH//32, 1, 1)
+    # the frame order after each denoising step will be 0,1,2 -> 2,0,1 -> 1,2,0 -> 0,1,2 ...
     modified_latents = torch.roll(unpacked_latents, shifts=1, dims=2)
     modified_latents = pipeline._pack_latents(modified_latents)
     return {"latents": modified_latents}
 
-@spaces.GPU(duration=120)
+@spaces.GPU(duration=140)
 def generate_gif(prompt, use_fixed_seed):
     seed = 0 if use_fixed_seed else torch.seed()
     generator = torch.Generator(device="cuda").manual_seed(seed)