smoothieAI commited on
Commit
8a2a238
·
verified ·
1 Parent(s): f75a7c9

Update pipeline.py

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