config.json

{
  "cond_dim": 1024,
  "depth": 12,
  "dim": 256,
  "expansion": 4,
  "input_dim": 3
}