smoothieAI commited on
Commit
eb50a45
·
verified ·
1 Parent(s): 365ad9a

Update pipeline.py

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