smoothieAI commited on
Commit
9319daa
·
verified ·
1 Parent(s): eb50a45

Update pipeline.py

Browse files
Files changed (1) hide show
  1. pipeline.py +3 -3
pipeline.py CHANGED
@@ -748,9 +748,9 @@ class AnimateDiffPipeline(DiffusionPipeline, TextualInversionLoaderMixin, IPAdap
748
  # scale the initial noise by the standard deviation required by the scheduler
749
  if init_latents is None:
750
  latents = latents * self.scheduler.init_noise_sigma
751
- elif self.unet.trained_initial_frames and init_latents is not None:
752
- # we only want to use this as the first frame
753
- init_latents[:, :, 1:] = torch.zeros_like(init_latents[:, :, 1:])
754
 
755
  latents = latents.to(device)
756
  return latents, init_latents
 
748
  # scale the initial noise by the standard deviation required by the scheduler
749
  if init_latents is None:
750
  latents = latents * self.scheduler.init_noise_sigma
751
+ # elif self.unet.trained_initial_frames and init_latents is not None:
752
+ # # we only want to use this as the first frame
753
+ # init_latents[:, :, 1:] = torch.zeros_like(init_latents[:, :, 1:])
754
 
755
  latents = latents.to(device)
756
  return latents, init_latents