sae-lens 6.15.0__tar.gz → 6.16.0__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 (40) hide show
  1. {sae_lens-6.15.0 → sae_lens-6.16.0}/PKG-INFO +1 -1
  2. {sae_lens-6.15.0 → sae_lens-6.16.0}/pyproject.toml +1 -1
  3. {sae_lens-6.15.0 → sae_lens-6.16.0}/sae_lens/__init__.py +1 -1
  4. {sae_lens-6.15.0 → sae_lens-6.16.0}/sae_lens/saes/sae.py +1 -0
  5. {sae_lens-6.15.0 → sae_lens-6.16.0}/sae_lens/training/sae_trainer.py +1 -0
  6. {sae_lens-6.15.0 → sae_lens-6.16.0}/LICENSE +0 -0
  7. {sae_lens-6.15.0 → sae_lens-6.16.0}/README.md +0 -0
  8. {sae_lens-6.15.0 → sae_lens-6.16.0}/sae_lens/analysis/__init__.py +0 -0
  9. {sae_lens-6.15.0 → sae_lens-6.16.0}/sae_lens/analysis/hooked_sae_transformer.py +0 -0
  10. {sae_lens-6.15.0 → sae_lens-6.16.0}/sae_lens/analysis/neuronpedia_integration.py +0 -0
  11. {sae_lens-6.15.0 → sae_lens-6.16.0}/sae_lens/cache_activations_runner.py +0 -0
  12. {sae_lens-6.15.0 → sae_lens-6.16.0}/sae_lens/config.py +0 -0
  13. {sae_lens-6.15.0 → sae_lens-6.16.0}/sae_lens/constants.py +0 -0
  14. {sae_lens-6.15.0 → sae_lens-6.16.0}/sae_lens/evals.py +0 -0
  15. {sae_lens-6.15.0 → sae_lens-6.16.0}/sae_lens/llm_sae_training_runner.py +0 -0
  16. {sae_lens-6.15.0 → sae_lens-6.16.0}/sae_lens/load_model.py +0 -0
  17. {sae_lens-6.15.0 → sae_lens-6.16.0}/sae_lens/loading/__init__.py +0 -0
  18. {sae_lens-6.15.0 → sae_lens-6.16.0}/sae_lens/loading/pretrained_sae_loaders.py +0 -0
  19. {sae_lens-6.15.0 → sae_lens-6.16.0}/sae_lens/loading/pretrained_saes_directory.py +0 -0
  20. {sae_lens-6.15.0 → sae_lens-6.16.0}/sae_lens/pretokenize_runner.py +0 -0
  21. {sae_lens-6.15.0 → sae_lens-6.16.0}/sae_lens/pretrained_saes.yaml +0 -0
  22. {sae_lens-6.15.0 → sae_lens-6.16.0}/sae_lens/registry.py +0 -0
  23. {sae_lens-6.15.0 → sae_lens-6.16.0}/sae_lens/saes/__init__.py +0 -0
  24. {sae_lens-6.15.0 → sae_lens-6.16.0}/sae_lens/saes/batchtopk_sae.py +0 -0
  25. {sae_lens-6.15.0 → sae_lens-6.16.0}/sae_lens/saes/gated_sae.py +0 -0
  26. {sae_lens-6.15.0 → sae_lens-6.16.0}/sae_lens/saes/jumprelu_sae.py +0 -0
  27. {sae_lens-6.15.0 → sae_lens-6.16.0}/sae_lens/saes/matryoshka_batchtopk_sae.py +0 -0
  28. {sae_lens-6.15.0 → sae_lens-6.16.0}/sae_lens/saes/standard_sae.py +0 -0
  29. {sae_lens-6.15.0 → sae_lens-6.16.0}/sae_lens/saes/topk_sae.py +0 -0
  30. {sae_lens-6.15.0 → sae_lens-6.16.0}/sae_lens/saes/transcoder.py +0 -0
  31. {sae_lens-6.15.0 → sae_lens-6.16.0}/sae_lens/tokenization_and_batching.py +0 -0
  32. {sae_lens-6.15.0 → sae_lens-6.16.0}/sae_lens/training/__init__.py +0 -0
  33. {sae_lens-6.15.0 → sae_lens-6.16.0}/sae_lens/training/activation_scaler.py +0 -0
  34. {sae_lens-6.15.0 → sae_lens-6.16.0}/sae_lens/training/activations_store.py +0 -0
  35. {sae_lens-6.15.0 → sae_lens-6.16.0}/sae_lens/training/mixing_buffer.py +0 -0
  36. {sae_lens-6.15.0 → sae_lens-6.16.0}/sae_lens/training/optim.py +0 -0
  37. {sae_lens-6.15.0 → sae_lens-6.16.0}/sae_lens/training/types.py +0 -0
  38. {sae_lens-6.15.0 → sae_lens-6.16.0}/sae_lens/training/upload_saes_to_huggingface.py +0 -0
  39. {sae_lens-6.15.0 → sae_lens-6.16.0}/sae_lens/tutorial/tsea.py +0 -0
  40. {sae_lens-6.15.0 → sae_lens-6.16.0}/sae_lens/util.py +0 -0
{sae_lens-6.15.0 → sae_lens-6.16.0}/PKG-INFO RENAMED
@@ -1,6 +1,6 @@
1
1
  Metadata-Version: 2.4
2
2
  Name: sae-lens
3
- Version: 6.15.0
3
+ Version: 6.16.0
4
4
  Summary: Training and Analyzing Sparse Autoencoders (SAEs)
5
5
  License: MIT
6
6
  License-File: LICENSE
{sae_lens-6.15.0 → sae_lens-6.16.0}/pyproject.toml RENAMED
@@ -1,6 +1,6 @@
1
1
  [tool.poetry]
2
2
  name = "sae-lens"
3
- version = "6.15.0"
3
+ version = "6.16.0"
4
4
  description = "Training and Analyzing Sparse Autoencoders (SAEs)"
5
5
  authors = ["Joseph Bloom"]
6
6
  readme = "README.md"
{sae_lens-6.15.0 → sae_lens-6.16.0}/sae_lens/__init__.py RENAMED
@@ -1,5 +1,5 @@
1
1
  # ruff: noqa: E402
2
- __version__ = "6.15.0"
2
+ __version__ = "6.16.0"
3
3
 
4
4
  import logging
5
5
 
{sae_lens-6.15.0 → sae_lens-6.16.0}/sae_lens/saes/sae.py RENAMED
@@ -217,6 +217,7 @@ class TrainStepInput:
217
217
  sae_in: torch.Tensor
218
218
  coefficients: dict[str, float]
219
219
  dead_neuron_mask: torch.Tensor | None
220
+ n_training_steps: int
220
221
 
221
222
 
222
223
  class TrainCoefficientConfig(NamedTuple):
{sae_lens-6.15.0 → sae_lens-6.16.0}/sae_lens/training/sae_trainer.py RENAMED
@@ -249,6 +249,7 @@ class SAETrainer(Generic[T_TRAINING_SAE, T_TRAINING_SAE_CONFIG]):
249
249
  sae_in=sae_in,
250
250
  dead_neuron_mask=self.dead_neurons,
251
251
  coefficients=self.get_coefficients(),
252
+ n_training_steps=self.n_training_steps,
252
253
  ),
253
254
  )
254
255
 
{sae_lens-6.15.0 → sae_lens-6.16.0}/LICENSE RENAMED
File without changes
{sae_lens-6.15.0 → sae_lens-6.16.0}/README.md RENAMED
File without changes
{sae_lens-6.15.0 → sae_lens-6.16.0}/sae_lens/config.py RENAMED
File without changes
{sae_lens-6.15.0 → sae_lens-6.16.0}/sae_lens/evals.py RENAMED
File without changes
{sae_lens-6.15.0 → sae_lens-6.16.0}/sae_lens/util.py RENAMED
File without changes