Use correct model name for np runner

jbloomAus · Apr 18, 2024 · 138d5d4 · 138d5d4
1 parent 6578436
commit 138d5d4
Showing 1 changed file with 1 addition and 1 deletion.
diff --git a/sae_lens/analysis/neuronpedia_runner.py b/sae_lens/analysis/neuronpedia_runner.py
@@ -88,7 +88,7 @@ def __init__(
         self.model, _, self.activation_store = (
             loader.load_sae_training_group_session()
         )
-        self.model_id = self.model.cfg.model_name
+        self.model_id = self.sparse_autoencoder.cfg.model_name
         self.layer = self.sparse_autoencoder.cfg.hook_point_layer
         self.sae_id = sae_id
         self.sparsity_threshold = sparsity_threshold