Update pipeline.py
Browse files- pipeline.py +6 -2
pipeline.py
CHANGED
@@ -611,6 +611,9 @@ class AnimateDiffPipeline(DiffusionPipeline, TextualInversionLoaderMixin, IPAdap
|
|
611 |
callback_steps: Optional[int] = 1,
|
612 |
cross_attention_kwargs: Optional[Dict[str, Any]] = None,
|
613 |
clip_skip: Optional[int] = None,
|
|
|
|
|
|
|
614 |
):
|
615 |
r"""
|
616 |
The call function to the pipeline for generation.
|
@@ -760,8 +763,9 @@ class AnimateDiffPipeline(DiffusionPipeline, TextualInversionLoaderMixin, IPAdap
|
|
760 |
device,
|
761 |
generator,
|
762 |
latents,
|
763 |
-
x_velocity=
|
764 |
-
y_velocity=
|
|
|
765 |
)
|
766 |
|
767 |
# 6. Prepare extra step kwargs. TODO: Logic should ideally just be moved out of the pipeline
|
|
|
611 |
callback_steps: Optional[int] = 1,
|
612 |
cross_attention_kwargs: Optional[Dict[str, Any]] = None,
|
613 |
clip_skip: Optional[int] = None,
|
614 |
+
x_velocity: Optional[float] = 0,
|
615 |
+
y_velocity: Optional[float] = 0,
|
616 |
+
scale_velocity: Optional[float] = 0,
|
617 |
):
|
618 |
r"""
|
619 |
The call function to the pipeline for generation.
|
|
|
763 |
device,
|
764 |
generator,
|
765 |
latents,
|
766 |
+
x_velocity=x_velocity,
|
767 |
+
y_velocity=y_velocity,
|
768 |
+
scale_velocity=scale_velocity,
|
769 |
)
|
770 |
|
771 |
# 6. Prepare extra step kwargs. TODO: Logic should ideally just be moved out of the pipeline
|