Skip to content

Commit 362f10a

Browse files
committed
Initial implementation of Dream Transformer model
1 parent 7ea065c commit 362f10a

4 files changed

Lines changed: 502 additions & 0 deletions

File tree

‎src/diffusers/__init__.py‎

Lines changed: 2 additions & 0 deletions
Original file line numberDiff line numberDiff line change
@@ -192,6 +192,7 @@
192192
"ControlNetXSAdapter",
193193
"CosmosTransformer3DModel",
194194
"DiTTransformer2DModel",
195+
"DreamTransformer1DModel",
195196
"EasyAnimateTransformer3DModel",
196197
"FluxControlNetModel",
197198
"FluxMultiControlNetModel",
@@ -853,6 +854,7 @@
853854
ControlNetXSAdapter,
854855
CosmosTransformer3DModel,
855856
DiTTransformer2DModel,
857+
DreamTransformer1DModel,
856858
EasyAnimateTransformer3DModel,
857859
FluxControlNetModel,
858860
FluxMultiControlNetModel,

‎src/diffusers/models/__init__.py‎

Lines changed: 2 additions & 0 deletions
Original file line numberDiff line numberDiff line change
@@ -80,6 +80,7 @@
8080
_import_structure["transformers.transformer_cogview3plus"] = ["CogView3PlusTransformer2DModel"]
8181
_import_structure["transformers.transformer_cogview4"] = ["CogView4Transformer2DModel"]
8282
_import_structure["transformers.transformer_cosmos"] = ["CosmosTransformer3DModel"]
83+
_import_structure["transformers.transformer_dream"] = ["DreamTransformer1DModel"]
8384
_import_structure["transformers.transformer_easyanimate"] = ["EasyAnimateTransformer3DModel"]
8485
_import_structure["transformers.transformer_flux"] = ["FluxTransformer2DModel"]
8586
_import_structure["transformers.transformer_hidream_image"] = ["HiDreamImageTransformer2DModel"]
@@ -165,6 +166,7 @@
165166
ConsisIDTransformer3DModel,
166167
CosmosTransformer3DModel,
167168
DiTTransformer2DModel,
169+
DreamTransformer1DModel,
168170
DualTransformer2DModel,
169171
EasyAnimateTransformer3DModel,
170172
FluxTransformer2DModel,

‎src/diffusers/models/transformers/__init__.py‎

Lines changed: 1 addition & 0 deletions
Original file line numberDiff line numberDiff line change
@@ -21,6 +21,7 @@
2121
from .transformer_cogview3plus import CogView3PlusTransformer2DModel
2222
from .transformer_cogview4 import CogView4Transformer2DModel
2323
from .transformer_cosmos import CosmosTransformer3DModel
24+
from .transformer_dream import DreamTransformer1DModel
2425
from .transformer_easyanimate import EasyAnimateTransformer3DModel
2526
from .transformer_flux import FluxTransformer2DModel
2627
from .transformer_hidream_image import HiDreamImageTransformer2DModel

0 commit comments

Comments
 (0)