config.json

{
  "model_id": "google/gemma-2-2b-it",
  "layer": 20,
  "num_blocks": 1,
  "n_steps": 3,
  "freeze_concept_enc": true,
  "disable_cross_attn": false,
  "disable_self_attn": false,
  "disable_mlp": false
}