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

Update pipeline.py

Browse files
Files changed (1) hide show
  1. pipeline.py +3 -1
pipeline.py CHANGED
@@ -50,6 +50,8 @@ from diffusers.pipelines.pipeline_utils import DiffusionPipeline
50
  from diffusers.utils import deprecate
51
 
52
  import torchvision
 
 
53
  import math
54
 
55
  logger = logging.get_logger(__name__) # pylint: disable=invalid-name
@@ -785,7 +787,7 @@ class AnimateDiffPipeline(DiffusionPipeline, TextualInversionLoaderMixin, IPAdap
785
  x_velocity: Optional[float] = 0,
786
  y_velocity: Optional[float] = 0,
787
  scale_velocity: Optional[float] = 0,
788
- init_image: Optional[str] = None,
789
  init_image_strength: Optional[float] = 1.0,
790
  init_noise_correlation: Optional[float] = 0.0,
791
  ):
 
50
  from diffusers.utils import deprecate
51
 
52
  import torchvision
53
+ import PIL
54
+ import PIL.Image
55
  import math
56
 
57
  logger = logging.get_logger(__name__) # pylint: disable=invalid-name
 
787
  x_velocity: Optional[float] = 0,
788
  y_velocity: Optional[float] = 0,
789
  scale_velocity: Optional[float] = 0,
790
+ init_image: Optional[PIL.Image.Image] = None,
791
  init_image_strength: Optional[float] = 1.0,
792
  init_noise_correlation: Optional[float] = 0.0,
793
  ):