Update pipeline.py
Browse files- 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[
|
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 |
):
|