SiT-S-2-256-diffusers/transformer/config.json

{
  "_class_name": "SiTTransformer2DModel",
  "_diffusers_version": "0.36.0",
  "class_dropout_prob": 0.1,
  "depth": 12,
  "hidden_size": 384,
  "in_channels": 4,
  "input_size": 32,
  "learn_sigma": true,
  "mlp_ratio": 4.0,
  "num_classes": 1000,
  "num_heads": 6,
  "patch_size": 2
}