Skip to content

Commit dc1d722

Browse files
committed
Introduce AutoPipelineForText2Video and AutoPipelineForImage2Video
1 parent 9352a5c commit dc1d722

File tree

4 files changed

+596
-0
lines changed

4 files changed

+596
-0
lines changed

src/diffusers/__init__.py

Lines changed: 4 additions & 0 deletions
Original file line numberDiff line numberDiff line change
@@ -229,6 +229,8 @@
229229
"AutoPipelineForImage2Image",
230230
"AutoPipelineForInpainting",
231231
"AutoPipelineForText2Image",
232+
"AutoPipelineForText2Video",
233+
"AutoPipelineForImage2Video",
232234
"ConsistencyModelPipeline",
233235
"DanceDiffusionPipeline",
234236
"DDIMPipeline",
@@ -818,8 +820,10 @@
818820
from .pipelines import (
819821
AudioPipelineOutput,
820822
AutoPipelineForImage2Image,
823+
AutoPipelineForImage2Video,
821824
AutoPipelineForInpainting,
822825
AutoPipelineForText2Image,
826+
AutoPipelineForText2Video,
823827
BlipDiffusionControlNetPipeline,
824828
BlipDiffusionPipeline,
825829
CLIPImageProjection,

src/diffusers/pipelines/__init__.py

Lines changed: 4 additions & 0 deletions
Original file line numberDiff line numberDiff line change
@@ -46,6 +46,8 @@
4646
"AutoPipelineForImage2Image",
4747
"AutoPipelineForInpainting",
4848
"AutoPipelineForText2Image",
49+
"AutoPipelineForText2Video",
50+
"AutoPipelineForImage2Video",
4951
]
5052
_import_structure["consistency_models"] = ["ConsistencyModelPipeline"]
5153
_import_structure["dance_diffusion"] = ["DanceDiffusionPipeline"]
@@ -468,8 +470,10 @@
468470
else:
469471
from .auto_pipeline import (
470472
AutoPipelineForImage2Image,
473+
AutoPipelineForImage2Video,
471474
AutoPipelineForInpainting,
472475
AutoPipelineForText2Image,
476+
AutoPipelineForText2Video,
473477
)
474478
from .consistency_models import ConsistencyModelPipeline
475479
from .dance_diffusion import DanceDiffusionPipeline

0 commit comments

Comments
 (0)