{
  "_class_name": "PoseDiTTransformer2DModel",
  "_diffusers_version": "0.32.1",
  "_name_or_path": "checkpoints/trainvaltest-vitbb-square-vertex-augv2-rot90-drop20-heatmap-8gpu-rect-normalize-flow-shift3-shape_aug-pretrain-zeroshape/20250127-115851/checkpoint-9000",
  "activation_fn": "gelu-approximate",
  "attention_bias": true,
  "attention_head_dim": 32,
  "dropout": 0.0,
  "in_channels": 3,
  "norm_elementwise_affine": false,
  "norm_eps": 1e-05,
  "norm_num_groups": 32,
  "norm_type": "ada_norm_zero",
  "num_attention_heads": 16,
  "num_embeds_ada_norm": 10,
  "num_joints": 283,
  "num_layers": 20,
  "out_channels": null,
  "patch_size": 2,
  "sample_size": 32,
  "upcast_attention": false,
  "use_heatmap": true
}