dit_model/config.json

{
  "_class_name": "DiT",
  "_diffusers_version": "0.32.2",
  "cross_attn_input_size": 4096,
  "depth": 40,
  "dynamic_softmax_temperature": false,
  "gradient_checkpoint": false,
  "hidden_size": 3072,
  "in_channels": 16,
  "mlp_ratio": 4.0,
  "num_heads": 12,
  "patch_size": 2,
  "residual_v": true,
  "rope_base": 10000,
  "train_bias_and_rms": false,
  "use_rope": true
}