smoothieAI commited on
Commit
511d871
·
verified ·
1 Parent(s): d041e6f

Update pipeline.py

Browse files
Files changed (1) hide show
  1. 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