transformer/config.json

{
  "_class_name": "ChromaTransformer2DModel",
  "_diffusers_version": "0.40.0.dev0",
  "approximator_hidden_dim": 32,
  "approximator_layers": 1,
  "approximator_num_channels": 16,
  "attention_head_dim": 16,
  "axes_dims_rope": [
    4,
    4,
    8
  ],
  "in_channels": 4,
  "joint_attention_dim": 32,
  "num_attention_heads": 2,
  "num_layers": 1,
  "num_single_layers": 1,
  "out_channels": null,
  "patch_size": 1
}