fix: allow tutorial packages for colab install to use latest version (#…

…173) fix: allow tutorial packages for colab install to use latest version (#173)
jbloomAus · Jun 5, 2024 · f73cb73 · f73cb73
1 parent 5d0faed
commit f73cb73
Show file tree

Hide file tree

Showing 3 changed files with 3 additions and 3 deletions.
diff --git a/tutorials/basic_loading_and_analysing.ipynb b/tutorials/basic_loading_and_analysing.ipynb
@@ -30,7 +30,7 @@
         "    import google.colab # type: ignore\n",
         "    from google.colab import output\n",
         "    COLAB = True\n",
-        "    %pip install sae-lens==1.3.0 transformer-lens==1.17.0\n",
+        "    %pip install sae-lens transformer-lens\n",
         "except:\n",
         "    COLAB = False\n",
         "    from IPython import get_ipython # type: ignore\n",

diff --git a/tutorials/logits_lens_with_features.ipynb b/tutorials/logits_lens_with_features.ipynb
@@ -53,7 +53,7 @@
         "    # For Google Colab, a high RAM instance is needed\n",
         "    import google.colab # type: ignore\n",
         "    from google.colab import output\n",
-        "    %pip install sae-lens==1.3.0 transformer-lens==1.17.0\n",
+        "    %pip install sae-lens transformer-lens\n",
         "except:\n",
         "    from IPython import get_ipython # type: ignore\n",
         "    ipython = get_ipython(); assert ipython is not None\n",

diff --git a/tutorials/training_a_sparse_autoencoder.ipynb b/tutorials/training_a_sparse_autoencoder.ipynb
@@ -31,7 +31,7 @@
         "try:\n",
         "    import google.colab # type: ignore\n",
         "    from google.colab import output\n",
-        "    %pip install sae-lens==1.3.0 transformer-lens==1.17.0 circuitsvis==1.43.2\n",
+        "    %pip install sae-lens transformer-lens circuitsvis\n",
         "except:\n",
         "    from IPython import get_ipython # type: ignore\n",
         "    ipython = get_ipython(); assert ipython is not None\n",