smoothieAI commited on
Commit
788e3dc
·
verified ·
1 Parent(s): d25ed14

Update pipeline.py

Browse files
Files changed (1) hide show
  1. pipeline.py +1 -6
pipeline.py CHANGED
@@ -1072,12 +1072,7 @@ class AnimateDiffPipeline(DiffusionPipeline, TextualInversionLoaderMixin, IPAdap
1072
  local_context_size = int(context_size * initial_context_scale)
1073
  for index in range(local_context_size):
1074
  # if its the first timestep, spread the indexes out evenly over the full frame range, offset by the group index
1075
- if timestep <= 1:
1076
- step_size = 2
1077
- frame_index = (group_index * (local_context_size - overlap)) + (offset * timestep) + index
1078
- else:
1079
- # Calculate the frame index
1080
- frame_index = (group_index * (local_context_size - overlap)) + (offset * timestep) + index
1081
  # If frame index exceeds total frames, wrap around
1082
  if frame_index >= total_frames:
1083
  frame_index %= total_frames
 
1072
  local_context_size = int(context_size * initial_context_scale)
1073
  for index in range(local_context_size):
1074
  # if its the first timestep, spread the indexes out evenly over the full frame range, offset by the group index
1075
+ frame_index = (group_index * (local_context_size - overlap)) + (offset * timestep) + index
 
 
 
 
 
1076
  # If frame index exceeds total frames, wrap around
1077
  if frame_index >= total_frames:
1078
  frame_index %= total_frames