Rename the source model directory (#94)

ai-safety-foundation · Nov 22, 2023 · e7bb30d · e7bb30d
1 parent f1c6381
commit e7bb30d
Show file tree

Hide file tree

Showing 4 changed files with 2 additions and 2 deletions.
diff --git a/sparse_autoencoder/src_model/__init__.py → sparse_autoencoder/source_model/__init__.py b/sparse_autoencoder/src_model/__init__.py → sparse_autoencoder/source_model/__init__.py
diff --git a/...coder/src_model/store_activations_hook.py → ...er/source_model/store_activations_hook.py b/...coder/src_model/store_activations_hook.py → ...er/source_model/store_activations_hook.py
diff --git a/...odel/tests/test_store_activations_hook.py → ...odel/tests/test_store_activations_hook.py b/...odel/tests/test_store_activations_hook.py → ...odel/tests/test_store_activations_hook.py
@@ -5,7 +5,7 @@
 from transformer_lens import HookedTransformer
 
 from sparse_autoencoder.activation_store.list_store import ListActivationStore
-from sparse_autoencoder.src_model.store_activations_hook import store_activations_hook
+from sparse_autoencoder.source_model.store_activations_hook import store_activations_hook
 from sparse_autoencoder.tensor_types import BatchTokenizedPrompts
 
 

diff --git a/sparse_autoencoder/train/pipeline.py b/sparse_autoencoder/train/pipeline.py
@@ -8,7 +8,7 @@
 
 from sparse_autoencoder.activation_store.tensor_store import TensorActivationStore
 from sparse_autoencoder.metrics.train.abstract_train_metric import TrainMetricData
-from sparse_autoencoder.src_model.store_activations_hook import store_activations_hook
+from sparse_autoencoder.source_model.store_activations_hook import store_activations_hook
 from sparse_autoencoder.tensor_types import BatchTokenizedPrompts, NeuronActivity
 from sparse_autoencoder.train.abstract_pipeline import AbstractPipeline
 from sparse_autoencoder.train.utils import get_model_device