smoothieAI commited on
Commit
26ad6f1
·
verified ·
1 Parent(s): 8c765c8

Update pipeline.py

Browse files
Files changed (1) hide show
  1. pipeline.py +1 -1
pipeline.py CHANGED
@@ -1004,7 +1004,7 @@ class AnimateDiffPipeline(DiffusionPipeline, TextualInversionLoaderMixin, IPAdap
1004
 
1005
 
1006
  # set the step index to the current batch
1007
- self.scheduler.step_index = i
1008
 
1009
  print(f"current_context_latents shape: {current_context_latents.shape}")
1010
  print(f"noise_pred shape: {noise_pred.shape}")
 
1004
 
1005
 
1006
  # set the step index to the current batch
1007
+ self.scheduler._step_index = i
1008
 
1009
  print(f"current_context_latents shape: {current_context_latents.shape}")
1010
  print(f"noise_pred shape: {noise_pred.shape}")