sae-lens 6.12.0__py3-none-any.whl → 6.12.2__py3-none-any.whl

sae_lens/__init__.py

@@ -1,5 +1,5 @@
 # ruff: noqa: E402
-__version__ = "6.12.0"
+__version__ = "6.12.2"
 
 import logging
 

sae_lens/pretrained_saes.yaml

@@ -14837,25 +14837,25 @@ llama-3.1-8b-instruct-andyrdt:
   saes:
   - id: resid_post_layer_3_trainer_1
     path: resid_post_layer_3/trainer_1
-    neuronpedia: llama3.1-8b-it/3-resid-post-ar
+    neuronpedia: llama3.1-8b-it/3-resid-post-aa
   - id: resid_post_layer_7_trainer_1
     path: resid_post_layer_7/trainer_1
-    neuronpedia: llama3.1-8b-it/7-resid-post-ar
+    neuronpedia: llama3.1-8b-it/7-resid-post-aa
   - id: resid_post_layer_11_trainer_1
     path: resid_post_layer_11/trainer_1
-    neuronpedia: llama3.1-8b-it/11-resid-post-ar
+    neuronpedia: llama3.1-8b-it/11-resid-post-aa
   - id: resid_post_layer_15_trainer_1
     path: resid_post_layer_15/trainer_1
-    neuronpedia: llama3.1-8b-it/15-resid-post-ar
+    neuronpedia: llama3.1-8b-it/15-resid-post-aa
   - id: resid_post_layer_19_trainer_1
     path: resid_post_layer_19/trainer_1
-    neuronpedia: llama3.1-8b-it/19-resid-post-ar
+    neuronpedia: llama3.1-8b-it/19-resid-post-aa
   - id: resid_post_layer_23_trainer_1
     path: resid_post_layer_23/trainer_1
-    neuronpedia: llama3.1-8b-it/23-resid-post-ar
+    neuronpedia: llama3.1-8b-it/23-resid-post-aa
   - id: resid_post_layer_27_trainer_1
     path: resid_post_layer_27/trainer_1
-    neuronpedia: llama3.1-8b-it/27-resid-post-ar
+    neuronpedia: llama3.1-8b-it/27-resid-post-aa
 
 qwen2.5-7b-instruct-andyrdt:
   conversion_func: dictionary_learning_1
@@ -14864,22 +14864,22 @@ qwen2.5-7b-instruct-andyrdt:
   saes:
   - id: resid_post_layer_3_trainer_1
     path: resid_post_layer_3/trainer_1
-    neuronpedia: qwen2.5-7b-it/3-resid-post-ar
+    neuronpedia: qwen2.5-7b-it/3-resid-post-aa
   - id: resid_post_layer_7_trainer_1
     path: resid_post_layer_7/trainer_1
-    neuronpedia: qwen2.5-7b-it/7-resid-post-ar
+    neuronpedia: qwen2.5-7b-it/7-resid-post-aa
   - id: resid_post_layer_11_trainer_1
     path: resid_post_layer_11/trainer_1
-    neuronpedia: qwen2.5-7b-it/11-resid-post-ar
+    neuronpedia: qwen2.5-7b-it/11-resid-post-aa
   - id: resid_post_layer_15_trainer_1
     path: resid_post_layer_15/trainer_1
-    neuronpedia: qwen2.5-7b-it/15-resid-post-ar
+    neuronpedia: qwen2.5-7b-it/15-resid-post-aa
   - id: resid_post_layer_19_trainer_1
     path: resid_post_layer_19/trainer_1
-    neuronpedia: qwen2.5-7b-it/19-resid-post-ar
+    neuronpedia: qwen2.5-7b-it/19-resid-post-aa
   - id: resid_post_layer_23_trainer_1
     path: resid_post_layer_23/trainer_1
-    neuronpedia: qwen2.5-7b-it/23-resid-post-ar
+    neuronpedia: qwen2.5-7b-it/23-resid-post-aa
   - id: resid_post_layer_27_trainer_1
     path: resid_post_layer_27/trainer_1
-    neuronpedia: qwen2.5-7b-it/27-resid-post-ar
+    neuronpedia: qwen2.5-7b-it/27-resid-post-aa

sae_lens/saes/sae.py

@@ -245,7 +245,7 @@ class SAE(HookedRootModule, Generic[T_SAE_CONFIG], ABC):
 
         self.cfg = cfg
 
-        if cfg.metadata and cfg.metadata:
+        if cfg.metadata and cfg.metadata.model_from_pretrained_kwargs:
             warnings.warn(
                 "\nThis SAE has non-empty model_from_pretrained_kwargs. "
                 "\nFor optimal performance, load the model like so:\n"

{sae_lens-6.12.0.dist-info → sae_lens-6.12.2.dist-info}/METADATA

@@ -1,8 +1,9 @@
-Metadata-Version: 2.3
+Metadata-Version: 2.4
 Name: sae-lens
-Version: 6.12.0
+Version: 6.12.2
 Summary: Training and Analyzing Sparse Autoencoders (SAEs)
 License: MIT
+License-File: LICENSE
 Keywords: deep-learning,sparse-autoencoders,mechanistic-interpretability,PyTorch
 Author: Joseph Bloom
 Requires-Python: >=3.10,<4.0
@@ -12,6 +13,7 @@ Classifier: Programming Language :: Python :: 3.10
 Classifier: Programming Language :: Python :: 3.11
 Classifier: Programming Language :: Python :: 3.12
 Classifier: Programming Language :: Python :: 3.13
+Classifier: Programming Language :: Python :: 3.14
 Classifier: Topic :: Scientific/Engineering :: Artificial Intelligence
 Provides-Extra: mamba
 Requires-Dist: babe (>=0.0.7,<0.0.8)

{sae_lens-6.12.0.dist-info → sae_lens-6.12.2.dist-info}/RECORD

@@ -1,4 +1,4 @@
-sae_lens/__init__.py,sha256=kTzPE14dVRpEIV0_yFC9whYCiJt6hoMQ33YhgwO83EI,3589
+sae_lens/__init__.py,sha256=TptOdqP3B6E_TTQ4n6DXAIDA9c1_9LUUsDkoqyrSSBg,3589
 sae_lens/analysis/__init__.py,sha256=47DEQpj8HBSa-_TImW-5JCeuQeRkm5NMpJWZG3hSuFU,0
 sae_lens/analysis/hooked_sae_transformer.py,sha256=vRu6JseH1lZaEeILD5bEkQEQ1wYHHDcxD-f2olKmE9Y,14275
 sae_lens/analysis/neuronpedia_integration.py,sha256=Gx1W7hUBEuMoasNcnOnZ1wmqbXDd1pSZ1nqKEya1HQc,4962
@@ -12,13 +12,13 @@ sae_lens/loading/__init__.py,sha256=47DEQpj8HBSa-_TImW-5JCeuQeRkm5NMpJWZG3hSuFU,
 sae_lens/loading/pretrained_sae_loaders.py,sha256=CVzHntSUKR1X3_gAqn8K_Ajq8D85qBrmrgEgU93IV4A,49609
 sae_lens/loading/pretrained_saes_directory.py,sha256=4Vn-Jex6SveD7EbxcSOBv8cx1gkPfUMLU1QOP-ww1ZE,3752
 sae_lens/pretokenize_runner.py,sha256=w0f6SfZLAxbp5eAAKnet8RqUB_DKofZ9RGsoJwFnYbA,7058
-sae_lens/pretrained_saes.yaml,sha256=n2BJVj8relA8OCbn9epJ5dFNg1_e9F9a4TouqjkXnts,602553
+sae_lens/pretrained_saes.yaml,sha256=6ca3geEB6NyhULUrmdtPDK8ea0YdpLp8_au78vIFC5w,602553
 sae_lens/registry.py,sha256=nhy7BPSudSATqW4lo9H_k3Na7sfGHmAf9v-3wpnLL_o,1490
 sae_lens/saes/__init__.py,sha256=jVwazK8Q6dW5J6_zFXPoNAuBvSxgziQ8eMOjGM3t-X8,1475
 sae_lens/saes/batchtopk_sae.py,sha256=GX_J0vH4vzeLqYxl0mkfsZQpFEoCEHMR4dIG8fz8N8w,3449
 sae_lens/saes/gated_sae.py,sha256=qcmM9JwBA8aZR8z_IRHV1_gQX-q_63tKewWXRnhdXuo,8986
 sae_lens/saes/jumprelu_sae.py,sha256=HHBF1sJ95lZvxwP5vwLSQFKdnJN2KKYK0WAEaLTrta0,13399
-sae_lens/saes/sae.py,sha256=gdUZuLaOHQrPjbDj-nZI813B6-_mNAnV9i9z4qTnpHk,38255
+sae_lens/saes/sae.py,sha256=McpF4pTh70r6SQUbHFm0YQ9X2c2qPULBUSd_YmnEk4Y,38284
 sae_lens/saes/standard_sae.py,sha256=9UqYyYtQuThYxXKNaDjYcyowpOx2-7cShG-TeUP6JCQ,5940
 sae_lens/saes/topk_sae.py,sha256=CXMBI6CFvI5829bOhoQ350VXR9d8uFHUDlULTIWHXoU,8686
 sae_lens/saes/transcoder.py,sha256=BfLSbTYVNZh-ruGxseZiZJ_acEL6_7QyTdfqUr0lDOg,12156
@@ -33,7 +33,7 @@ sae_lens/training/types.py,sha256=qSjmGzXf3MLalygG0psnVjmhX_mpLmL47MQtZfe7qxg,81
 sae_lens/training/upload_saes_to_huggingface.py,sha256=r_WzI1zLtGZ5TzAxuG3xa_8T09j3zXJrWd_vzPsPGkQ,4469
 sae_lens/tutorial/tsea.py,sha256=fd1am_XXsf2KMbByDapJo-2qlxduKaa62Z2qcQZ3QKU,18145
 sae_lens/util.py,sha256=lW7fBn_b8quvRYlen9PUmB7km60YhKyjmuelB1f6KzQ,2253
-sae_lens-6.12.0.dist-info/LICENSE,sha256=DW6e-hDosiu4CfW0-imI57sV1I5f9UEslpviNQcOAKs,1069
-sae_lens-6.12.0.dist-info/METADATA,sha256=6L9rhKmaK8oQdRZ9-p0TxqpbBXCaY4bcwKqHP_NYnEw,5245
-sae_lens-6.12.0.dist-info/WHEEL,sha256=b4K_helf-jlQoXBBETfwnf4B04YC67LOev0jo4fX5m8,88
-sae_lens-6.12.0.dist-info/RECORD,,
+sae_lens-6.12.2.dist-info/METADATA,sha256=m8hF8tj-b70b5iAvN21ZDxOXzRHRxDmwpJclZzHqPw4,5318
+sae_lens-6.12.2.dist-info/WHEEL,sha256=zp0Cn7JsFoX2ATtOhtaFYIiE2rmFAD4OcMhtUki8W3U,88
+sae_lens-6.12.2.dist-info/licenses/LICENSE,sha256=DW6e-hDosiu4CfW0-imI57sV1I5f9UEslpviNQcOAKs,1069
+sae_lens-6.12.2.dist-info/RECORD,,

{sae_lens-6.12.0.dist-info → sae_lens-6.12.2.dist-info}/WHEEL

@@ -1,4 +1,4 @@
 Wheel-Version: 1.0
-Generator: poetry-core 2.1.3
+Generator: poetry-core 2.2.1
 Root-Is-Purelib: true
 Tag: py3-none-any