Skip to content

Commit 1891d8d

Browse files
committed
Add inital commit for InfiniteTalk Transformer (based on Wan I2V transformer)
1 parent 5c3c1ac commit 1891d8d

File tree

4 files changed

+1161
-0
lines changed

4 files changed

+1161
-0
lines changed

src/diffusers/__init__.py

Lines changed: 2 additions & 0 deletions
Original file line numberDiff line numberDiff line change
@@ -218,6 +218,7 @@
218218
"HunyuanVideoFramepackTransformer3DModel",
219219
"HunyuanVideoTransformer3DModel",
220220
"I2VGenXLUNet",
221+
"InfiniteTalkTransformer3DModel",
221222
"Kandinsky3UNet",
222223
"LatteTransformer3DModel",
223224
"LTXVideoTransformer3DModel",
@@ -898,6 +899,7 @@
898899
HunyuanVideoFramepackTransformer3DModel,
899900
HunyuanVideoTransformer3DModel,
900901
I2VGenXLUNet,
902+
InfiniteTalkTransformer3DModel,
901903
Kandinsky3UNet,
902904
LatteTransformer3DModel,
903905
LTXVideoTransformer3DModel,

src/diffusers/models/__init__.py

Lines changed: 2 additions & 0 deletions
Original file line numberDiff line numberDiff line change
@@ -90,6 +90,7 @@
9090
_import_structure["transformers.transformer_hidream_image"] = ["HiDreamImageTransformer2DModel"]
9191
_import_structure["transformers.transformer_hunyuan_video"] = ["HunyuanVideoTransformer3DModel"]
9292
_import_structure["transformers.transformer_hunyuan_video_framepack"] = ["HunyuanVideoFramepackTransformer3DModel"]
93+
_import_structure["transformers.transformer_infinitetalk"] = ["InfiniteTalkTransformer3DModel"]
9394
_import_structure["transformers.transformer_ltx"] = ["LTXVideoTransformer3DModel"]
9495
_import_structure["transformers.transformer_lumina2"] = ["Lumina2Transformer2DModel"]
9596
_import_structure["transformers.transformer_mochi"] = ["MochiTransformer3DModel"]
@@ -180,6 +181,7 @@
180181
HunyuanDiT2DModel,
181182
HunyuanVideoFramepackTransformer3DModel,
182183
HunyuanVideoTransformer3DModel,
184+
InfiniteTalkTransformer3DModel,
183185
LatteTransformer3DModel,
184186
LTXVideoTransformer3DModel,
185187
Lumina2Transformer2DModel,

src/diffusers/models/transformers/__init__.py

Lines changed: 1 addition & 0 deletions
Original file line numberDiff line numberDiff line change
@@ -27,6 +27,7 @@
2727
from .transformer_hidream_image import HiDreamImageTransformer2DModel
2828
from .transformer_hunyuan_video import HunyuanVideoTransformer3DModel
2929
from .transformer_hunyuan_video_framepack import HunyuanVideoFramepackTransformer3DModel
30+
from .transformer_infinitetalk import InfiniteTalkTransformer3DModel
3031
from .transformer_ltx import LTXVideoTransformer3DModel
3132
from .transformer_lumina2 import Lumina2Transformer2DModel
3233
from .transformer_mochi import MochiTransformer3DModel

0 commit comments

Comments
 (0)