pMF-L-32/transformer/config.json

{
  "_class_name": "PMFTransformer2DModel",
  "_diffusers_version": "0.38.0",
  "aux_head_depth": 8,
  "bottleneck_dim": 128,
  "depth": 32,
  "embedding_init_constant": 1.0,
  "eval_mode": true,
  "hidden_size": 1024,
  "in_channels": 3,
  "mlp_ratio": 2.6666666666666665,
  "model_type": "pMF-L/32",
  "norm_eps": 1e-06,
  "num_attention_heads": 16,
  "num_cfg_tokens": 4,
  "num_class_embeds": null,
  "num_class_tokens": 8,
  "num_classes": 1000,
  "num_interval_tokens": 2,
  "num_time_tokens": 4,
  "patch_size": 32,
  "sample_size": 512,
  "t_clip_min": 0.05,
  "token_init_constant": 1.0,
  "weight_init_constant": 0.32
}