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