smoothieAI
/

pipeline_animatediff_context

Model card Files Files and versions Community

smoothieAI commited on Jan 18, 2024

Commit

ccbc4b4

·

verified ·

1 Parent(s): 5549023

Update pipeline.py

Files changed (1) hide show

pipeline.py +1 -1

pipeline.py CHANGED Viewed

@@ -1008,7 +1008,7 @@ class AnimateDiffPipeline(DiffusionPipeline, TextualInversionLoaderMixin, IPAdap
         def context_scheduler(context_size, overlap, offset, total_frames, total_timesteps):
             # Calculate the number of context groups based on frame count and context size
-            number_of_context_groups = (total_frames // (context_size - overlap)) + 1
             # Initialize a list to store context indexes for all timesteps
             all_context_indexes = []
             # Iterate over each timestep

         def context_scheduler(context_size, overlap, offset, total_frames, total_timesteps):
             # Calculate the number of context groups based on frame count and context size
+            number_of_context_groups = (total_frames // (context_size - overlap))
             # Initialize a list to store context indexes for all timesteps
             all_context_indexes = []
             # Iterate over each timestep