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

Update pipeline.py

Browse files
Files changed (1) hide show
  1. pipeline.py +4 -1
pipeline.py CHANGED
@@ -1002,7 +1002,10 @@ class AnimateDiffPipeline(DiffusionPipeline, TextualInversionLoaderMixin, IPAdap
1002
  noise_pred_uncond, noise_pred_text = noise_pred.chunk(2)
1003
  noise_pred = noise_pred_uncond + guidance_scale * (noise_pred_text - noise_pred_uncond)
1004
 
1005
-
 
 
 
1006
  print(f"current_context_latents shape: {current_context_latents.shape}")
1007
  print(f"noise_pred shape: {noise_pred.shape}")
1008
 
 
1002
  noise_pred_uncond, noise_pred_text = noise_pred.chunk(2)
1003
  noise_pred = noise_pred_uncond + guidance_scale * (noise_pred_text - noise_pred_uncond)
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}")
1011