Update pipeline.py
Browse files- pipeline.py +4 -0
pipeline.py
CHANGED
@@ -1004,6 +1004,10 @@ class AnimateDiffPipeline(DiffusionPipeline, TextualInversionLoaderMixin, IPAdap
|
|
1004 |
print(f"current_context_latents shape: {current_context_latents.shape}")
|
1005 |
print(f"noise_pred shape: {noise_pred.shape}")
|
1006 |
|
|
|
|
|
|
|
|
|
1007 |
# compute the previous noisy sample x_t -> x_t-1
|
1008 |
current_context_latents = self.scheduler.step(noise_pred, t, current_context_latents, **extra_step_kwargs).prev_sample
|
1009 |
|
|
|
1004 |
print(f"current_context_latents shape: {current_context_latents.shape}")
|
1005 |
print(f"noise_pred shape: {noise_pred.shape}")
|
1006 |
|
1007 |
+
|
1008 |
+
# print size of sigmas
|
1009 |
+
print(f"sigma len: {len(self.scheduler.sigmas)}")
|
1010 |
+
|
1011 |
# compute the previous noisy sample x_t -> x_t-1
|
1012 |
current_context_latents = self.scheduler.step(noise_pred, t, current_context_latents, **extra_step_kwargs).prev_sample
|
1013 |
|