Skip to content

Commit ec068f9

Browse files
authored
fix dual transformer2d import (#8491)
fix
1 parent 0240d41 commit ec068f9

File tree

1 file changed

+1
-1
lines changed

1 file changed

+1
-1
lines changed

src/diffusers/models/__init__.py

Lines changed: 1 addition & 1 deletion
Original file line numberDiff line numberDiff line change
@@ -34,10 +34,10 @@
3434
_import_structure["autoencoders.vq_model"] = ["VQModel"]
3535
_import_structure["controlnet"] = ["ControlNetModel"]
3636
_import_structure["controlnet_xs"] = ["ControlNetXSAdapter", "UNetControlNetXSModel"]
37-
_import_structure["dual_transformer_2d"] = ["DualTransformer2DModel"]
3837
_import_structure["embeddings"] = ["ImageProjection"]
3938
_import_structure["modeling_utils"] = ["ModelMixin"]
4039
_import_structure["transformers.dit_transformer_2d"] = ["DiTTransformer2DModel"]
40+
_import_structure["transformers.dual_transformer_2d"] = ["DualTransformer2DModel"]
4141
_import_structure["transformers.hunyuan_transformer_2d"] = ["HunyuanDiT2DModel"]
4242
_import_structure["transformers.pixart_transformer_2d"] = ["PixArtTransformer2DModel"]
4343
_import_structure["transformers.prior_transformer"] = ["PriorTransformer"]

0 commit comments

Comments
 (0)