Skip to content

Commit 02f8c26

Browse files
committed
refactor pt. 1
1 parent 98602d8 commit 02f8c26

File tree

5 files changed

+108
-187
lines changed

5 files changed

+108
-187
lines changed

src/diffusers/models/__init__.py

Lines changed: 1 addition & 1 deletion
Original file line numberDiff line numberDiff line change
@@ -60,7 +60,6 @@
6060
_import_structure["transformers.consisid_transformer_3d"] = ["ConsisIDTransformer3DModel"]
6161
_import_structure["transformers.dit_transformer_2d"] = ["DiTTransformer2DModel"]
6262
_import_structure["transformers.dual_transformer_2d"] = ["DualTransformer2DModel"]
63-
_import_structure["transformers.easyanimate_transformer_3d"] = ["EasyAnimateTransformer3DModel"]
6463
_import_structure["transformers.hunyuan_transformer_2d"] = ["HunyuanDiT2DModel"]
6564
_import_structure["transformers.latte_transformer_3d"] = ["LatteTransformer3DModel"]
6665
_import_structure["transformers.lumina_nextdit2d"] = ["LuminaNextDiT2DModel"]
@@ -72,6 +71,7 @@
7271
_import_structure["transformers.transformer_2d"] = ["Transformer2DModel"]
7372
_import_structure["transformers.transformer_allegro"] = ["AllegroTransformer3DModel"]
7473
_import_structure["transformers.transformer_cogview3plus"] = ["CogView3PlusTransformer2DModel"]
74+
_import_structure["transformers.transformer_easyanimate"] = ["EasyAnimateTransformer3DModel"]
7575
_import_structure["transformers.transformer_flux"] = ["FluxTransformer2DModel"]
7676
_import_structure["transformers.transformer_hunyuan_video"] = ["HunyuanVideoTransformer3DModel"]
7777
_import_structure["transformers.transformer_ltx"] = ["LTXVideoTransformer3DModel"]

0 commit comments

Comments
 (0)