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