smoothieAI commited on
Commit
b0f31b3
·
verified ·
1 Parent(s): 01b8dc6

Update pipeline.py

Browse files
Files changed (1) hide show
  1. 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=0.1,
764
- y_velocity=0.1,
 
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