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