smoothieAI commited on
Commit
9fc4b8c
·
verified ·
1 Parent(s): 8547fc1

Update pipeline.py

Browse files
Files changed (1) hide show
  1. pipeline.py +3 -0
pipeline.py CHANGED
@@ -151,6 +151,9 @@ class AnimateDiffPipeline(DiffusionPipeline, TextualInversionLoaderMixin, IPAdap
151
  self.vae_scale_factor = 2 ** (len(self.vae.config.block_out_channels) - 1)
152
  self.image_processor = VaeImageProcessor(vae_scale_factor=self.vae_scale_factor)
153
 
 
 
 
154
  # Copied from diffusers.pipelines.stable_diffusion.pipeline_stable_diffusion.StableDiffusionPipeline.encode_prompt with num_images_per_prompt -> num_videos_per_prompt
155
  def encode_prompt(
156
  self,
 
151
  self.vae_scale_factor = 2 ** (len(self.vae.config.block_out_channels) - 1)
152
  self.image_processor = VaeImageProcessor(vae_scale_factor=self.vae_scale_factor)
153
 
154
+ def load_motion_adapter(self,motion_adapter):
155
+ self.unet.load_motion_modules(motion_adapter)
156
+
157
  # Copied from diffusers.pipelines.stable_diffusion.pipeline_stable_diffusion.StableDiffusionPipeline.encode_prompt with num_images_per_prompt -> num_videos_per_prompt
158
  def encode_prompt(
159
  self,