| { | |
| "_class_name": "JoyImageEditTransformer3DModel", | |
| "_diffusers_version": "0.38.0.dev0", | |
| "attn_backend": "torch_spda", | |
| "dit_modulation_type": "wanx", | |
| "enable_activation_checkpointing": false, | |
| "heads_num": 32, | |
| "hidden_size": 4096, | |
| "in_channels": 16, | |
| "is_repa": false, | |
| "mlp_width_ratio": 4.0, | |
| "mm_double_blocks_depth": 40, | |
| "out_channels": 16, | |
| "patch_size": [ | |
| 1, | |
| 2, | |
| 2 | |
| ], | |
| "repa_layer": 13, | |
| "rope_dim_list": [ | |
| 16, | |
| 56, | |
| 56 | |
| ], | |
| "rope_theta": 10000, | |
| "rope_type": "rope", | |
| "text_states_dim": 4096, | |
| "unpatchify_new": true | |
| } | |