smoothieAI
/

pipeline_animatediff_context

Model card Files Files and versions Community

smoothieAI commited on Jan 12, 2024

Commit

0359c79

·

verified ·

1 Parent(s): 9fc4b8c

Update pipeline.py

Files changed (1) hide show

pipeline.py +1 -1

pipeline.py CHANGED Viewed

@@ -152,7 +152,7 @@ class AnimateDiffPipeline(DiffusionPipeline, TextualInversionLoaderMixin, IPAdap
         self.image_processor = VaeImageProcessor(vae_scale_factor=self.vae_scale_factor)
     def load_motion_adapter(self,motion_adapter):
-        self.unet.load_motion_modules(motion_adapter)
     # Copied from diffusers.pipelines.stable_diffusion.pipeline_stable_diffusion.StableDiffusionPipeline.encode_prompt with num_images_per_prompt -> num_videos_per_prompt
     def encode_prompt(

         self.image_processor = VaeImageProcessor(vae_scale_factor=self.vae_scale_factor)
     def load_motion_adapter(self,motion_adapter):
+        self.register_modules(motion_adapter=motion_adapter)
     # Copied from diffusers.pipelines.stable_diffusion.pipeline_stable_diffusion.StableDiffusionPipeline.encode_prompt with num_images_per_prompt -> num_videos_per_prompt
     def encode_prompt(