Update pipeline.py
Browse files- 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,
|