transformer/config.json

{
  "_class_name": "PartragDiTModel",
  "_diffusers_version": "0.35.1",
  "_name_or_path": "michaelpopo/PartRAG",
  "cross_attention_dim": 1024,
  "decay": 0.9999,
  "enable_global_cross_attn": true,
  "enable_local_cross_attn": true,
  "enable_part_embedding": true,
  "global_attn_block_id_range": null,
  "global_attn_block_ids": [
    0,
    2,
    4,
    6,
    8,
    10,
    12,
    14,
    16,
    18,
    20
  ],
  "in_channels": 64,
  "inv_gamma": 1.0,
  "max_num_parts": 32,
  "min_decay": 0.0,
  "num_attention_heads": 16,
  "num_layers": 21,
  "optimization_step": 120,
  "power": 0.75,
  "update_after_step": 0,
  "use_ema_warmup": true,
  "width": 2048
}