|
31 | 31 | _import_structure["autoencoders.autoencoder_kl"] = ["AutoencoderKL"] |
32 | 32 | _import_structure["autoencoders.autoencoder_kl_allegro"] = ["AutoencoderKLAllegro"] |
33 | 33 | _import_structure["autoencoders.autoencoder_kl_cogvideox"] = ["AutoencoderKLCogVideoX"] |
34 | | - _import_structure["autoencoders.autoencoder_kl_magvit"] = ["AutoencoderKLMagvit"] |
35 | 34 | _import_structure["autoencoders.autoencoder_kl_hunyuan_video"] = ["AutoencoderKLHunyuanVideo"] |
36 | 35 | _import_structure["autoencoders.autoencoder_kl_ltx"] = ["AutoencoderKLLTXVideo"] |
| 36 | + _import_structure["autoencoders.autoencoder_kl_magvit"] = ["AutoencoderKLMagvit"] |
37 | 37 | _import_structure["autoencoders.autoencoder_kl_mochi"] = ["AutoencoderKLMochi"] |
38 | 38 | _import_structure["autoencoders.autoencoder_kl_temporal_decoder"] = ["AutoencoderKLTemporalDecoder"] |
39 | 39 | _import_structure["autoencoders.autoencoder_oobleck"] = ["AutoencoderOobleck"] |
|
57 | 57 | _import_structure["modeling_utils"] = ["ModelMixin"] |
58 | 58 | _import_structure["transformers.auraflow_transformer_2d"] = ["AuraFlowTransformer2DModel"] |
59 | 59 | _import_structure["transformers.cogvideox_transformer_3d"] = ["CogVideoXTransformer3DModel"] |
60 | | - _import_structure["transformers.easyanimate_transformer_3d"] = ["EasyAnimateTransformer3DModel"] |
61 | 60 | _import_structure["transformers.consisid_transformer_3d"] = ["ConsisIDTransformer3DModel"] |
62 | 61 | _import_structure["transformers.dit_transformer_2d"] = ["DiTTransformer2DModel"] |
63 | 62 | _import_structure["transformers.dual_transformer_2d"] = ["DualTransformer2DModel"] |
| 63 | + _import_structure["transformers.easyanimate_transformer_3d"] = ["EasyAnimateTransformer3DModel"] |
64 | 64 | _import_structure["transformers.hunyuan_transformer_2d"] = ["HunyuanDiT2DModel"] |
65 | 65 | _import_structure["transformers.latte_transformer_3d"] = ["LatteTransformer3DModel"] |
66 | 66 | _import_structure["transformers.lumina_nextdit2d"] = ["LuminaNextDiT2DModel"] |
|
0 commit comments