Skip to content

Commit c9a219b

Browse files
fancydaddyDN6
fancydaddy
andauthored
add from_single_file to animatediff (#10924)
* Update pipeline_animatediff.py * Update pipeline_animatediff_controlnet.py * Update pipeline_animatediff_sparsectrl.py * Update pipeline_animatediff_video2video.py * Update pipeline_animatediff_video2video_controlnet.py --------- Co-authored-by: Dhruv Nair <dhruv.nair@gmail.com>
1 parent 9e910c4 commit c9a219b

File tree

5 files changed

+10
-5
lines changed

5 files changed

+10
-5
lines changed

src/diffusers/pipelines/animatediff/pipeline_animatediff.py

Lines changed: 2 additions & 1 deletion
Original file line numberDiff line numberDiff line change
@@ -19,7 +19,7 @@
1919
from transformers import CLIPImageProcessor, CLIPTextModel, CLIPTokenizer, CLIPVisionModelWithProjection
2020

2121
from ...image_processor import PipelineImageInput
22-
from ...loaders import IPAdapterMixin, StableDiffusionLoraLoaderMixin, TextualInversionLoaderMixin
22+
from ...loaders import FromSingleFileMixin, IPAdapterMixin, StableDiffusionLoraLoaderMixin, TextualInversionLoaderMixin
2323
from ...models import AutoencoderKL, ImageProjection, UNet2DConditionModel, UNetMotionModel
2424
from ...models.lora import adjust_lora_scale_text_encoder
2525
from ...models.unets.unet_motion_model import MotionAdapter
@@ -83,6 +83,7 @@ class AnimateDiffPipeline(
8383
StableDiffusionLoraLoaderMixin,
8484
FreeInitMixin,
8585
AnimateDiffFreeNoiseMixin,
86+
FromSingleFileMixin,
8687
):
8788
r"""
8889
Pipeline for text-to-video generation.

src/diffusers/pipelines/animatediff/pipeline_animatediff_controlnet.py

Lines changed: 2 additions & 1 deletion
Original file line numberDiff line numberDiff line change
@@ -20,7 +20,7 @@
2020
from transformers import CLIPImageProcessor, CLIPTextModel, CLIPTokenizer, CLIPVisionModelWithProjection
2121

2222
from ...image_processor import PipelineImageInput
23-
from ...loaders import IPAdapterMixin, StableDiffusionLoraLoaderMixin, TextualInversionLoaderMixin
23+
from ...loaders import FromSingleFileMixin, IPAdapterMixin, StableDiffusionLoraLoaderMixin, TextualInversionLoaderMixin
2424
from ...models import (
2525
AutoencoderKL,
2626
ControlNetModel,
@@ -125,6 +125,7 @@ class AnimateDiffControlNetPipeline(
125125
StableDiffusionLoraLoaderMixin,
126126
FreeInitMixin,
127127
AnimateDiffFreeNoiseMixin,
128+
FromSingleFileMixin,
128129
):
129130
r"""
130131
Pipeline for text-to-video generation with ControlNet guidance.

src/diffusers/pipelines/animatediff/pipeline_animatediff_sparsectrl.py

Lines changed: 2 additions & 1 deletion
Original file line numberDiff line numberDiff line change
@@ -22,7 +22,7 @@
2222
from transformers import CLIPImageProcessor, CLIPTextModel, CLIPTokenizer, CLIPVisionModelWithProjection
2323

2424
from ...image_processor import PipelineImageInput, VaeImageProcessor
25-
from ...loaders import IPAdapterMixin, StableDiffusionLoraLoaderMixin, TextualInversionLoaderMixin
25+
from ...loaders import FromSingleFileMixin, IPAdapterMixin, StableDiffusionLoraLoaderMixin, TextualInversionLoaderMixin
2626
from ...models import AutoencoderKL, ImageProjection, UNet2DConditionModel, UNetMotionModel
2727
from ...models.controlnets.controlnet_sparsectrl import SparseControlNetModel
2828
from ...models.lora import adjust_lora_scale_text_encoder
@@ -136,6 +136,7 @@ class AnimateDiffSparseControlNetPipeline(
136136
IPAdapterMixin,
137137
StableDiffusionLoraLoaderMixin,
138138
FreeInitMixin,
139+
FromSingleFileMixin,
139140
):
140141
r"""
141142
Pipeline for controlled text-to-video generation using the method described in [SparseCtrl: Adding Sparse Controls

src/diffusers/pipelines/animatediff/pipeline_animatediff_video2video.py

Lines changed: 2 additions & 1 deletion
Original file line numberDiff line numberDiff line change
@@ -19,7 +19,7 @@
1919
from transformers import CLIPImageProcessor, CLIPTextModel, CLIPTokenizer, CLIPVisionModelWithProjection
2020

2121
from ...image_processor import PipelineImageInput
22-
from ...loaders import IPAdapterMixin, StableDiffusionLoraLoaderMixin, TextualInversionLoaderMixin
22+
from ...loaders import FromSingleFileMixin, IPAdapterMixin, StableDiffusionLoraLoaderMixin, TextualInversionLoaderMixin
2323
from ...models import AutoencoderKL, ImageProjection, UNet2DConditionModel, UNetMotionModel
2424
from ...models.lora import adjust_lora_scale_text_encoder
2525
from ...models.unets.unet_motion_model import MotionAdapter
@@ -186,6 +186,7 @@ class AnimateDiffVideoToVideoPipeline(
186186
StableDiffusionLoraLoaderMixin,
187187
FreeInitMixin,
188188
AnimateDiffFreeNoiseMixin,
189+
FromSingleFileMixin,
189190
):
190191
r"""
191192
Pipeline for video-to-video generation.

src/diffusers/pipelines/animatediff/pipeline_animatediff_video2video_controlnet.py

Lines changed: 2 additions & 1 deletion
Original file line numberDiff line numberDiff line change
@@ -20,7 +20,7 @@
2020
from transformers import CLIPImageProcessor, CLIPTextModel, CLIPTokenizer, CLIPVisionModelWithProjection
2121

2222
from ...image_processor import PipelineImageInput
23-
from ...loaders import IPAdapterMixin, StableDiffusionLoraLoaderMixin, TextualInversionLoaderMixin
23+
from ...loaders import FromSingleFileMixin, IPAdapterMixin, StableDiffusionLoraLoaderMixin, TextualInversionLoaderMixin
2424
from ...models import (
2525
AutoencoderKL,
2626
ControlNetModel,
@@ -204,6 +204,7 @@ class AnimateDiffVideoToVideoControlNetPipeline(
204204
StableDiffusionLoraLoaderMixin,
205205
FreeInitMixin,
206206
AnimateDiffFreeNoiseMixin,
207+
FromSingleFileMixin,
207208
):
208209
r"""
209210
Pipeline for video-to-video generation with ControlNet guidance.

0 commit comments

Comments
 (0)