sae-lens 6.28.0__tar.gz → 6.28.1__tar.gz

{sae_lens-6.28.0 → sae_lens-6.28.1}/PKG-INFO

@@ -1,6 +1,6 @@
 Metadata-Version: 2.4
 Name: sae-lens
-Version: 6.28.0
+Version: 6.28.1
 Summary: Training and Analyzing Sparse Autoencoders (SAEs)
 License: MIT
 License-File: LICENSE

{sae_lens-6.28.0 → sae_lens-6.28.1}/pyproject.toml

@@ -1,6 +1,6 @@
 [tool.poetry]
 name = "sae-lens"
-version = "6.28.0"
+version = "6.28.1"
 description = "Training and Analyzing Sparse Autoencoders (SAEs)"
 authors = ["Joseph Bloom"]
 readme = "README.md"

{sae_lens-6.28.0 → sae_lens-6.28.1}/sae_lens/__init__.py

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

{sae_lens-6.28.0 → sae_lens-6.28.1}/sae_lens/loading/pretrained_saes_directory.py

@@ -103,3 +103,21 @@ def get_config_overrides(release: str, sae_id: str) -> dict[str, Any]:
         if sae_info.neuronpedia_id is not None and sae_id in sae_info.neuronpedia_id:
             config_overrides["neuronpedia_id"] = sae_info.neuronpedia_id[sae_id]
     return config_overrides
+
+
+def get_releases_for_repo_id(repo_id: str) -> list[str]:
+    """
+    Find all release names that use the given HuggingFace repo_id.
+
+    Args:
+        repo_id: The HuggingFace repository ID to search for.
+
+    Returns:
+        A list of release names that use this repo_id.
+    """
+    saes_directory = get_pretrained_saes_directory()
+    return [
+        release
+        for release, lookup in saes_directory.items()
+        if lookup.repo_id == repo_id
+    ]

{sae_lens-6.28.0 → sae_lens-6.28.1}/sae_lens/saes/sae.py

@@ -47,6 +47,7 @@ from sae_lens.loading.pretrained_saes_directory import (
     get_config_overrides,
     get_norm_scaling_factor,
     get_pretrained_saes_directory,
+    get_releases_for_repo_id,
     get_repo_id_and_folder_name,
 )
 from sae_lens.registry import get_sae_class, get_sae_training_class
@@ -624,6 +625,18 @@ class SAE(HookedRootModule, Generic[T_SAE_CONFIG], ABC):
                 raise ValueError(
                     f"Release {release} not found in pretrained SAEs directory, and is not a valid huggingface repo."
                 )
+            # Check if the user passed a repo_id that's in the pretrained SAEs list
+            matching_releases = get_releases_for_repo_id(release)
+            if matching_releases:
+                warnings.warn(
+                    f"You are loading an SAE using the HuggingFace repo_id '{release}' directly. "
+                    f"This repo is registered in the official pretrained SAEs list with release name(s): {matching_releases}. "
+                    f"For better compatibility and to access additional metadata, consider loading with: "
+                    f"SAE.from_pretrained(release='{matching_releases[0]}', sae_id='<sae_id>'). "
+                    f"See the full list of pretrained SAEs at: https://decoderesearch.github.io/SAELens/latest/pretrained_saes/",
+                    UserWarning,
+                    stacklevel=2,
+                )
         elif sae_id not in sae_directory[release].saes_map:
             # Handle special cases like Gemma Scope
             if (