sae-lens 6.5.2__tar.gz → 6.5.3__tar.gz

This diff represents the content of publicly available package versions that have been released to one of the supported registries. The information contained in this diff is provided for informational purposes only and reflects changes between package versions as they appear in their respective public registries.
Files changed (39) hide show
  1. {sae_lens-6.5.2 → sae_lens-6.5.3}/PKG-INFO +1 -1
  2. {sae_lens-6.5.2 → sae_lens-6.5.3}/pyproject.toml +1 -1
  3. {sae_lens-6.5.2 → sae_lens-6.5.3}/sae_lens/__init__.py +1 -1
  4. {sae_lens-6.5.2 → sae_lens-6.5.3}/sae_lens/loading/pretrained_sae_loaders.py +1 -0
  5. {sae_lens-6.5.2 → sae_lens-6.5.3}/sae_lens/saes/transcoder.py +5 -0
  6. {sae_lens-6.5.2 → sae_lens-6.5.3}/LICENSE +0 -0
  7. {sae_lens-6.5.2 → sae_lens-6.5.3}/README.md +0 -0
  8. {sae_lens-6.5.2 → sae_lens-6.5.3}/sae_lens/analysis/__init__.py +0 -0
  9. {sae_lens-6.5.2 → sae_lens-6.5.3}/sae_lens/analysis/hooked_sae_transformer.py +0 -0
  10. {sae_lens-6.5.2 → sae_lens-6.5.3}/sae_lens/analysis/neuronpedia_integration.py +0 -0
  11. {sae_lens-6.5.2 → sae_lens-6.5.3}/sae_lens/cache_activations_runner.py +0 -0
  12. {sae_lens-6.5.2 → sae_lens-6.5.3}/sae_lens/config.py +0 -0
  13. {sae_lens-6.5.2 → sae_lens-6.5.3}/sae_lens/constants.py +0 -0
  14. {sae_lens-6.5.2 → sae_lens-6.5.3}/sae_lens/evals.py +0 -0
  15. {sae_lens-6.5.2 → sae_lens-6.5.3}/sae_lens/llm_sae_training_runner.py +0 -0
  16. {sae_lens-6.5.2 → sae_lens-6.5.3}/sae_lens/load_model.py +0 -0
  17. {sae_lens-6.5.2 → sae_lens-6.5.3}/sae_lens/loading/__init__.py +0 -0
  18. {sae_lens-6.5.2 → sae_lens-6.5.3}/sae_lens/loading/pretrained_saes_directory.py +0 -0
  19. {sae_lens-6.5.2 → sae_lens-6.5.3}/sae_lens/pretokenize_runner.py +0 -0
  20. {sae_lens-6.5.2 → sae_lens-6.5.3}/sae_lens/pretrained_saes.yaml +0 -0
  21. {sae_lens-6.5.2 → sae_lens-6.5.3}/sae_lens/registry.py +0 -0
  22. {sae_lens-6.5.2 → sae_lens-6.5.3}/sae_lens/saes/__init__.py +0 -0
  23. {sae_lens-6.5.2 → sae_lens-6.5.3}/sae_lens/saes/batchtopk_sae.py +0 -0
  24. {sae_lens-6.5.2 → sae_lens-6.5.3}/sae_lens/saes/gated_sae.py +0 -0
  25. {sae_lens-6.5.2 → sae_lens-6.5.3}/sae_lens/saes/jumprelu_sae.py +0 -0
  26. {sae_lens-6.5.2 → sae_lens-6.5.3}/sae_lens/saes/sae.py +0 -0
  27. {sae_lens-6.5.2 → sae_lens-6.5.3}/sae_lens/saes/standard_sae.py +0 -0
  28. {sae_lens-6.5.2 → sae_lens-6.5.3}/sae_lens/saes/topk_sae.py +0 -0
  29. {sae_lens-6.5.2 → sae_lens-6.5.3}/sae_lens/tokenization_and_batching.py +0 -0
  30. {sae_lens-6.5.2 → sae_lens-6.5.3}/sae_lens/training/__init__.py +0 -0
  31. {sae_lens-6.5.2 → sae_lens-6.5.3}/sae_lens/training/activation_scaler.py +0 -0
  32. {sae_lens-6.5.2 → sae_lens-6.5.3}/sae_lens/training/activations_store.py +0 -0
  33. {sae_lens-6.5.2 → sae_lens-6.5.3}/sae_lens/training/mixing_buffer.py +0 -0
  34. {sae_lens-6.5.2 → sae_lens-6.5.3}/sae_lens/training/optim.py +0 -0
  35. {sae_lens-6.5.2 → sae_lens-6.5.3}/sae_lens/training/sae_trainer.py +0 -0
  36. {sae_lens-6.5.2 → sae_lens-6.5.3}/sae_lens/training/types.py +0 -0
  37. {sae_lens-6.5.2 → sae_lens-6.5.3}/sae_lens/training/upload_saes_to_huggingface.py +0 -0
  38. {sae_lens-6.5.2 → sae_lens-6.5.3}/sae_lens/tutorial/tsea.py +0 -0
  39. {sae_lens-6.5.2 → sae_lens-6.5.3}/sae_lens/util.py +0 -0
{sae_lens-6.5.2 → sae_lens-6.5.3}/PKG-INFO RENAMED
@@ -1,6 +1,6 @@
1
1
  Metadata-Version: 2.3
2
2
  Name: sae-lens
3
- Version: 6.5.2
3
+ Version: 6.5.3
4
4
  Summary: Training and Analyzing Sparse Autoencoders (SAEs)
5
5
  License: MIT
6
6
  Keywords: deep-learning,sparse-autoencoders,mechanistic-interpretability,PyTorch
{sae_lens-6.5.2 → sae_lens-6.5.3}/pyproject.toml RENAMED
@@ -1,6 +1,6 @@
1
1
  [tool.poetry]
2
2
  name = "sae-lens"
3
- version = "6.5.2"
3
+ version = "6.5.3"
4
4
  description = "Training and Analyzing Sparse Autoencoders (SAEs)"
5
5
  authors = ["Joseph Bloom"]
6
6
  readme = "README.md"
{sae_lens-6.5.2 → sae_lens-6.5.3}/sae_lens/__init__.py RENAMED
@@ -1,5 +1,5 @@
1
1
  # ruff: noqa: E402
2
- __version__ = "6.5.2"
2
+ __version__ = "6.5.3"
3
3
 
4
4
  import logging
5
5
 
{sae_lens-6.5.2 → sae_lens-6.5.3}/sae_lens/loading/pretrained_sae_loaders.py RENAMED
@@ -1177,6 +1177,7 @@ def get_gemma_2_transcoder_config_from_hf(
1177
1177
  "prepend_bos": True,
1178
1178
  "dataset_path": "monology/pile-uncopyrighted",
1179
1179
  "context_size": 1024,
1180
+ "apply_b_dec_to_input": False,
1180
1181
  **(cfg_overrides or {}),
1181
1182
  }
1182
1183
 
{sae_lens-6.5.2 → sae_lens-6.5.3}/sae_lens/saes/transcoder.py RENAMED
@@ -52,6 +52,11 @@ class TranscoderConfig(SAEConfig):
52
52
 
53
53
  return res
54
54
 
55
+ def __post_init__(self):
56
+ if self.apply_b_dec_to_input:
57
+ raise ValueError("apply_b_dec_to_input is not supported for transcoders")
58
+ return super().__post_init__()
59
+
55
60
 
56
61
  class Transcoder(SAE[TranscoderConfig]):
57
62
  """
{sae_lens-6.5.2 → sae_lens-6.5.3}/LICENSE RENAMED
File without changes
{sae_lens-6.5.2 → sae_lens-6.5.3}/README.md RENAMED
File without changes
{sae_lens-6.5.2 → sae_lens-6.5.3}/sae_lens/config.py RENAMED
File without changes
{sae_lens-6.5.2 → sae_lens-6.5.3}/sae_lens/constants.py RENAMED
File without changes
{sae_lens-6.5.2 → sae_lens-6.5.3}/sae_lens/evals.py RENAMED
File without changes
{sae_lens-6.5.2 → sae_lens-6.5.3}/sae_lens/registry.py RENAMED
File without changes
{sae_lens-6.5.2 → sae_lens-6.5.3}/sae_lens/saes/sae.py RENAMED
File without changes
{sae_lens-6.5.2 → sae_lens-6.5.3}/sae_lens/util.py RENAMED
File without changes