AFM-B-2-1NFE-guided/generator/config.json

{
  "_class_name": "DiTTransformer2DModel",
  "_diffusers_version": "0.36.0",
  "sample_size": 32,
  "num_layers": 12,
  "num_attention_heads": 12,
  "attention_head_dim": 64,
  "in_channels": 4,
  "out_channels": 4,
  "patch_size": 2,
  "attention_bias": true,
  "activation_fn": "gelu-approximate",
  "num_embeds_ada_norm": 1000,
  "norm_type": "ada_norm_zero",
  "norm_elementwise_affine": false,
  "dropout": 0.0,
  "norm_num_groups": 32,
  "norm_eps": 1e-05,
  "upcast_attention": false,
  "model_type": "AFM-B/2",
  "architecture": "standard",
  "repeat": 1,
  "use_t_src": false,
  "use_t_tgt": false,
  "pred_type": "x",
  "num_inference_steps": 1,
  "learn_sigma": false,
  "class_dropout_prob": 0.0,
  "input_size": 32,
  "num_classes": 1000,
  "depth": 12,
  "hidden_size": 768,
  "num_heads": 12
}