Update pipeline.py

pipeline.py

@@ -725,30 +725,30 @@ class AnimateDiffPipeline(DiffusionPipeline, TextualInversionLoaderMixin, IPAdap
                 latent_sum = torch.zeros_like(latents).to(device).to(dtype=torch.float16)
                 latent_counter = torch.zeros(num_frames).to(device).to(dtype=torch.float16)
         
-                # # foreach context group seperately denoise the current timestep
-                # for context_group in range(num_context_groups):
-                #     # calculate to current indexes, considering overlap
-                #     if context_group == 0:current_context_start = 0
-                #     else:current_context_start = context_group * (context_size - overlap)
-
-                #     # select the relevent context from the latents
-                #     current_context_latents = latents[:, :, current_context_start : current_context_start + context_size, :, :]
-                #     # if the context extends past the end of the latents, wrap around to the start
-                #     if current_context_start + context_size > num_frames:
-                #         current_context_latents = torch.cat([current_context_latents, latents[:, :, :current_context_start + context_size - num_frames, :, :]], dim=2)
-                    
+                # foreach context group seperately denoise the current timestep
                 for context_group in range(num_context_groups):
-                    # Calculate the current start index, considering overlap
-                    current_context_start = 0 if context_group == 0 else context_group * (context_size - overlap)
+                    # calculate to current indexes, considering overlap
+                    if context_group == 0:current_context_start = 0
+                    else:current_context_start = context_group * (context_size - overlap)
+
+                    # select the relevent context from the latents
+                    current_context_latents = latents[:, :, current_context_start : current_context_start + context_size, :, :]
+                    # if the context extends past the end of the latents, wrap around to the start
+                    if current_context_start + context_size > num_frames:
+                        current_context_latents = torch.cat([current_context_latents, latents[:, :, :current_context_start + context_size - num_frames, :, :]], dim=2)
+                    
+                # for context_group in range(num_context_groups):
+                #     # Calculate the current start index, considering overlap
+                #     current_context_start = 0 if context_group == 0 else context_group * (context_size - overlap)
 
-                    # Calculate the end index and adjust if it exceeds num_frames
-                    current_context_end = (current_context_start + context_size) % num_frames
+                #     # Calculate the end index and adjust if it exceeds num_frames
+                #     current_context_end = (current_context_start + context_size) % num_frames
 
-                    # Select the relevant context from the latents with wrap-around handling
-                    current_context_latents = torch.cat([
-                        latents[:, :, current_context_start:min(current_context_start + context_size, num_frames), :, :],
-                        latents[:, :, :max(current_context_end - num_frames, 0), :, :]
-                    ], dim=2) if current_context_start + context_size > num_frames else latents[:, :, current_context_start:current_context_start + context_size, :, :]
+                #     # Select the relevant context from the latents with wrap-around handling
+                #     current_context_latents = torch.cat([
+                #         latents[:, :, current_context_start:min(current_context_start + context_size, num_frames), :, :],
+                #         latents[:, :, :max(current_context_end - num_frames, 0), :, :]
+                #     ], dim=2) if current_context_start + context_size > num_frames else latents[:, :, current_context_start:current_context_start + context_size, :, :]