{
  "dim": 1024,
  "depth": 22,
  "heads": 16,
  "ff_mult": 2,
  "text_dim": 512,
  "conv_layers": 4
}