Skip to content

Commit 154e150

Browse files
author
XCLiu
committed
Hunyuan Team: add HunyuanDiT related updates
1 parent baab065 commit 154e150

File tree

9 files changed

+2015
-0
lines changed

9 files changed

+2015
-0
lines changed

src/diffusers/__init__.py

Lines changed: 4 additions & 0 deletions
Original file line numberDiff line numberDiff line change
@@ -82,6 +82,7 @@
8282
"ConsistencyDecoderVAE",
8383
"ControlNetModel",
8484
"ControlNetXSAdapter",
85+
"HunyuanDiT2DModel",
8586
"I2VGenXLUNet",
8687
"Kandinsky3UNet",
8788
"ModelMixin",
@@ -227,6 +228,7 @@
227228
"BlipDiffusionPipeline",
228229
"CLIPImageProjection",
229230
"CycleDiffusionPipeline",
231+
"HunyuanDiTPipeline",
230232
"I2VGenXLPipeline",
231233
"IFImg2ImgPipeline",
232234
"IFImg2ImgSuperResolutionPipeline",
@@ -482,6 +484,7 @@
482484
ConsistencyDecoderVAE,
483485
ControlNetModel,
484486
ControlNetXSAdapter,
487+
HunyuanDiT2DModel,
485488
I2VGenXLUNet,
486489
Kandinsky3UNet,
487490
ModelMixin,
@@ -605,6 +608,7 @@
605608
AudioLDMPipeline,
606609
CLIPImageProjection,
607610
CycleDiffusionPipeline,
611+
HunyuanDiTPipeline,
608612
I2VGenXLPipeline,
609613
IFImg2ImgPipeline,
610614
IFImg2ImgSuperResolutionPipeline,

src/diffusers/models/__init__.py

Lines changed: 2 additions & 0 deletions
Original file line numberDiff line numberDiff line change
@@ -39,6 +39,7 @@
3939
_import_structure["transformers.prior_transformer"] = ["PriorTransformer"]
4040
_import_structure["transformers.t5_film_transformer"] = ["T5FilmDecoder"]
4141
_import_structure["transformers.transformer_2d"] = ["Transformer2DModel"]
42+
_import_structure["transformers.hunyuan_transformer_2d"] = ["HunyuanDiT2DModel"]
4243
_import_structure["transformers.transformer_temporal"] = ["TransformerTemporalModel"]
4344
_import_structure["unets.unet_1d"] = ["UNet1DModel"]
4445
_import_structure["unets.unet_2d"] = ["UNet2DModel"]
@@ -73,6 +74,7 @@
7374
from .embeddings import ImageProjection
7475
from .modeling_utils import ModelMixin
7576
from .transformers import (
77+
HunyuanDiT2DModel,
7678
DualTransformer2DModel,
7779
PriorTransformer,
7880
T5FilmDecoder,

src/diffusers/models/attention.py

Lines changed: 414 additions & 0 deletions
Large diffs are not rendered by default.

src/diffusers/models/transformers/__init__.py

Lines changed: 1 addition & 0 deletions
Original file line numberDiff line numberDiff line change
@@ -7,3 +7,4 @@
77
from .t5_film_transformer import T5FilmDecoder
88
from .transformer_2d import Transformer2DModel
99
from .transformer_temporal import TransformerTemporalModel
10+
from .hunyuan_transformer_2d import HunyuanDiT2DModel

0 commit comments

Comments
 (0)