Update pipeline.py
Browse files- 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 |
|