e2_model_config.json

{
  "dim": 1024,
  "depth": 24,
  "heads": 16,
  "ff_mult": 4
}