| name: dit3d | |
| variant: full | |
| pos_emb_type: rope_3d | |
| patch_size: 2 | |
| hidden_size: 384 | |
| depth: 12 | |
| num_heads: 6 | |
| mlp_ratio: 4.0 | |
| use_gradient_checkpointing: False |
| name: dit3d | |
| variant: full | |
| pos_emb_type: rope_3d | |
| patch_size: 2 | |
| hidden_size: 384 | |
| depth: 12 | |
| num_heads: 6 | |
| mlp_ratio: 4.0 | |
| use_gradient_checkpointing: False |