config.json

{
  "model_type": "query2sae",
  "backbone_name": "gpt2",
  "head_hidden_dim": 128,
  "sae_dim": 16384,
  "architectures": ["Query2SAEModel"],
  "auto_map": {
    "AutoConfig": "configuration_query2sae.Query2SAEConfig",
    "AutoModel":  "modeling_query2sae.Query2SAEModel"
  }
}