layer_2/config.json

{
  "base_model": "esm2_t6_8M_UR50D",
  "layer": 2,
  "model_type": "sparse_autoencoder",
  "architecture": {
    "esm_dim": 320,
    "expansion_factor": 32,
    "feature_dim": 10240
  }
}