Skip to content

Commit 5c3c1ac

Browse files
committed
Add initial commit for InfiniteTalk pipeline based on the Wav I2V pipeline
1 parent c2e5ece commit 5c3c1ac

File tree

5 files changed

+906
-0
lines changed

5 files changed

+906
-0
lines changed

src/diffusers/__init__.py

Lines changed: 2 additions & 0 deletions
Original file line numberDiff line numberDiff line change
@@ -462,6 +462,7 @@
462462
"IFPipeline",
463463
"IFSuperResolutionPipeline",
464464
"ImageTextPipelineOutput",
465+
"InfiniteTalkPipeline",
465466
"Kandinsky3Img2ImgPipeline",
466467
"Kandinsky3Pipeline",
467468
"KandinskyCombinedPipeline",
@@ -1111,6 +1112,7 @@
11111112
IFPipeline,
11121113
IFSuperResolutionPipeline,
11131114
ImageTextPipelineOutput,
1115+
InfiniteTalkPipeline,
11141116
Kandinsky3Img2ImgPipeline,
11151117
Kandinsky3Pipeline,
11161118
KandinskyCombinedPipeline,

src/diffusers/pipelines/__init__.py

Lines changed: 2 additions & 0 deletions
Original file line numberDiff line numberDiff line change
@@ -240,6 +240,7 @@
240240
"HunyuanVideoImageToVideoPipeline",
241241
"HunyuanVideoFramepackPipeline",
242242
]
243+
_import_structure["infinitetalk"] = "InfiniteTalkPipeline"
243244
_import_structure["kandinsky"] = [
244245
"KandinskyCombinedPipeline",
245246
"KandinskyImg2ImgCombinedPipeline",
@@ -643,6 +644,7 @@
643644
)
644645
from .hunyuandit import HunyuanDiTPipeline
645646
from .i2vgen_xl import I2VGenXLPipeline
647+
from .infinitetalk import InfiniteTalkPipeline
646648
from .kandinsky import (
647649
KandinskyCombinedPipeline,
648650
KandinskyImg2ImgCombinedPipeline,
Lines changed: 47 additions & 0 deletions
Original file line numberDiff line numberDiff line change
@@ -0,0 +1,47 @@
1+
from typing import TYPE_CHECKING
2+
3+
from ...utils import (
4+
DIFFUSERS_SLOW_IMPORT,
5+
OptionalDependencyNotAvailable,
6+
_LazyModule,
7+
get_objects_from_module,
8+
is_torch_available,
9+
is_transformers_available,
10+
)
11+
12+
13+
_dummy_objects = {}
14+
_import_structure = {"pipeline_output": ["InfiniteTalkPipelineOutput"]}
15+
16+
17+
try:
18+
if not (is_transformers_available() and is_torch_available()):
19+
raise OptionalDependencyNotAvailable()
20+
except OptionalDependencyNotAvailable:
21+
from ...utils import dummy_torch_and_transformers_objects # noqa F403
22+
23+
_dummy_objects.update(get_objects_from_module(dummy_torch_and_transformers_objects))
24+
else:
25+
_import_structure["pipeline_infinitetalk"] = ["InfiniteTalkPipeline"]
26+
if TYPE_CHECKING or DIFFUSERS_SLOW_IMPORT:
27+
try:
28+
if not (is_transformers_available() and is_torch_available()):
29+
raise OptionalDependencyNotAvailable()
30+
31+
except OptionalDependencyNotAvailable:
32+
from ...utils.dummy_torch_and_transformers_objects import *
33+
else:
34+
from .pipeline_infinitetalk import InfiniteTalkPipeline
35+
36+
else:
37+
import sys
38+
39+
sys.modules[__name__] = _LazyModule(
40+
__name__,
41+
globals()["__file__"],
42+
_import_structure,
43+
module_spec=__spec__,
44+
)
45+
46+
for name, value in _dummy_objects.items():
47+
setattr(sys.modules[__name__], name, value)

0 commit comments

Comments
 (0)