sae-lens 6.24.1__tar.gz → 6.24.2__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 (41) hide show
  1. {sae_lens-6.24.1 → sae_lens-6.24.2}/PKG-INFO +1 -1
  2. {sae_lens-6.24.1 → sae_lens-6.24.2}/pyproject.toml +1 -1
  3. {sae_lens-6.24.1 → sae_lens-6.24.2}/sae_lens/__init__.py +1 -1
  4. {sae_lens-6.24.1 → sae_lens-6.24.2}/sae_lens/loading/pretrained_sae_loaders.py +0 -9
  5. {sae_lens-6.24.1 → sae_lens-6.24.2}/LICENSE +0 -0
  6. {sae_lens-6.24.1 → sae_lens-6.24.2}/README.md +0 -0
  7. {sae_lens-6.24.1 → sae_lens-6.24.2}/sae_lens/analysis/__init__.py +0 -0
  8. {sae_lens-6.24.1 → sae_lens-6.24.2}/sae_lens/analysis/hooked_sae_transformer.py +0 -0
  9. {sae_lens-6.24.1 → sae_lens-6.24.2}/sae_lens/analysis/neuronpedia_integration.py +0 -0
  10. {sae_lens-6.24.1 → sae_lens-6.24.2}/sae_lens/cache_activations_runner.py +0 -0
  11. {sae_lens-6.24.1 → sae_lens-6.24.2}/sae_lens/config.py +0 -0
  12. {sae_lens-6.24.1 → sae_lens-6.24.2}/sae_lens/constants.py +0 -0
  13. {sae_lens-6.24.1 → sae_lens-6.24.2}/sae_lens/evals.py +0 -0
  14. {sae_lens-6.24.1 → sae_lens-6.24.2}/sae_lens/llm_sae_training_runner.py +0 -0
  15. {sae_lens-6.24.1 → sae_lens-6.24.2}/sae_lens/load_model.py +0 -0
  16. {sae_lens-6.24.1 → sae_lens-6.24.2}/sae_lens/loading/__init__.py +0 -0
  17. {sae_lens-6.24.1 → sae_lens-6.24.2}/sae_lens/loading/pretrained_saes_directory.py +0 -0
  18. {sae_lens-6.24.1 → sae_lens-6.24.2}/sae_lens/pretokenize_runner.py +0 -0
  19. {sae_lens-6.24.1 → sae_lens-6.24.2}/sae_lens/pretrained_saes.yaml +0 -0
  20. {sae_lens-6.24.1 → sae_lens-6.24.2}/sae_lens/registry.py +0 -0
  21. {sae_lens-6.24.1 → sae_lens-6.24.2}/sae_lens/saes/__init__.py +0 -0
  22. {sae_lens-6.24.1 → sae_lens-6.24.2}/sae_lens/saes/batchtopk_sae.py +0 -0
  23. {sae_lens-6.24.1 → sae_lens-6.24.2}/sae_lens/saes/gated_sae.py +0 -0
  24. {sae_lens-6.24.1 → sae_lens-6.24.2}/sae_lens/saes/jumprelu_sae.py +0 -0
  25. {sae_lens-6.24.1 → sae_lens-6.24.2}/sae_lens/saes/matryoshka_batchtopk_sae.py +0 -0
  26. {sae_lens-6.24.1 → sae_lens-6.24.2}/sae_lens/saes/sae.py +0 -0
  27. {sae_lens-6.24.1 → sae_lens-6.24.2}/sae_lens/saes/standard_sae.py +0 -0
  28. {sae_lens-6.24.1 → sae_lens-6.24.2}/sae_lens/saes/temporal_sae.py +0 -0
  29. {sae_lens-6.24.1 → sae_lens-6.24.2}/sae_lens/saes/topk_sae.py +0 -0
  30. {sae_lens-6.24.1 → sae_lens-6.24.2}/sae_lens/saes/transcoder.py +0 -0
  31. {sae_lens-6.24.1 → sae_lens-6.24.2}/sae_lens/tokenization_and_batching.py +0 -0
  32. {sae_lens-6.24.1 → sae_lens-6.24.2}/sae_lens/training/__init__.py +0 -0
  33. {sae_lens-6.24.1 → sae_lens-6.24.2}/sae_lens/training/activation_scaler.py +0 -0
  34. {sae_lens-6.24.1 → sae_lens-6.24.2}/sae_lens/training/activations_store.py +0 -0
  35. {sae_lens-6.24.1 → sae_lens-6.24.2}/sae_lens/training/mixing_buffer.py +0 -0
  36. {sae_lens-6.24.1 → sae_lens-6.24.2}/sae_lens/training/optim.py +0 -0
  37. {sae_lens-6.24.1 → sae_lens-6.24.2}/sae_lens/training/sae_trainer.py +0 -0
  38. {sae_lens-6.24.1 → sae_lens-6.24.2}/sae_lens/training/types.py +0 -0
  39. {sae_lens-6.24.1 → sae_lens-6.24.2}/sae_lens/training/upload_saes_to_huggingface.py +0 -0
  40. {sae_lens-6.24.1 → sae_lens-6.24.2}/sae_lens/tutorial/tsea.py +0 -0
  41. {sae_lens-6.24.1 → sae_lens-6.24.2}/sae_lens/util.py +0 -0
{sae_lens-6.24.1 → sae_lens-6.24.2}/PKG-INFO RENAMED
@@ -1,6 +1,6 @@
1
1
  Metadata-Version: 2.4
2
2
  Name: sae-lens
3
- Version: 6.24.1
3
+ Version: 6.24.2
4
4
  Summary: Training and Analyzing Sparse Autoencoders (SAEs)
5
5
  License: MIT
6
6
  License-File: LICENSE
{sae_lens-6.24.1 → sae_lens-6.24.2}/pyproject.toml RENAMED
@@ -1,6 +1,6 @@
1
1
  [tool.poetry]
2
2
  name = "sae-lens"
3
- version = "6.24.1"
3
+ version = "6.24.2"
4
4
  description = "Training and Analyzing Sparse Autoencoders (SAEs)"
5
5
  authors = ["Joseph Bloom"]
6
6
  readme = "README.md"
{sae_lens-6.24.1 → sae_lens-6.24.2}/sae_lens/__init__.py RENAMED
@@ -1,5 +1,5 @@
1
1
  # ruff: noqa: E402
2
- __version__ = "6.24.1"
2
+ __version__ = "6.24.2"
3
3
 
4
4
  import logging
5
5
 
{sae_lens-6.24.1 → sae_lens-6.24.2}/sae_lens/loading/pretrained_sae_loaders.py RENAMED
@@ -682,15 +682,6 @@ def gemma_3_sae_huggingface_loader(
682
682
  cfg_overrides,
683
683
  )
684
684
 
685
- # replace folder name of 65k with 64k
686
- # TODO: remove this workaround once weights are fixed
687
- if "270m-pt" in repo_id:
688
- if "65k" in folder_name:
689
- folder_name = folder_name.replace("65k", "64k")
690
- # replace folder name of 262k with 250k
691
- if "262k" in folder_name:
692
- folder_name = folder_name.replace("262k", "250k")
693
-
694
685
  params_file = "params.safetensors"
695
686
  if "clt" in folder_name:
696
687
  params_file = folder_name.split("/")[-1] + ".safetensors"
{sae_lens-6.24.1 → sae_lens-6.24.2}/LICENSE RENAMED
File without changes
{sae_lens-6.24.1 → sae_lens-6.24.2}/README.md RENAMED
File without changes
{sae_lens-6.24.1 → sae_lens-6.24.2}/sae_lens/config.py RENAMED
File without changes
{sae_lens-6.24.1 → sae_lens-6.24.2}/sae_lens/evals.py RENAMED
File without changes
{sae_lens-6.24.1 → sae_lens-6.24.2}/sae_lens/util.py RENAMED
File without changes