Update pipeline.py
Browse files- pipeline.py +1 -1
pipeline.py
CHANGED
@@ -667,7 +667,7 @@ class AnimateDiffPipeline(DiffusionPipeline, TextualInversionLoaderMixin, IPAdap
|
|
667 |
)
|
668 |
|
669 |
if init_image is not None:
|
670 |
-
start_image = ((torchvision.transforms.functional.pil_to_tensor(init_image))/ 255 )[:3, :, :].to("cuda").to(torch.bfloat16).unsqueeze(0)
|
671 |
start_image = (
|
672 |
self.vae.encode(start_image.mul(2).sub(1))
|
673 |
.latent_dist.sample()
|
|
|
667 |
)
|
668 |
|
669 |
if init_image is not None:
|
670 |
+
start_image = ((torchvision.transforms.functional.pil_to_tensor(init_image))/ 255 )[:3, :, :].to("cuda").to(torch.bfloat16).unsqueeze(0)
|
671 |
start_image = (
|
672 |
self.vae.encode(start_image.mul(2).sub(1))
|
673 |
.latent_dist.sample()
|