🎨 add default versions for pip install in google colab

Author: subercui

tutorials/zero-shot/Tutorial_ZeroShot_Integration.ipynb

@@ -51,15 +51,17 @@
     "\n",
     "import os\n",
     "import sys\n",
+    "%load_ext autoreload\n",
+    "%autoreload 2\n",
     "\n",
     "if \"google.colab\" in sys.modules:\n",
     "    print(\"Running on Google Colab\")\n",
     "    print(\"Installing dependencies...\")\n",
-    "    !pip install -U scgpt \"torch<=2.2.2\"\n",
+    "    !pip install -U scgpt \"torch<=2.2.2\" \"numpy<2\" \"umap-learn<0.5.7\"\n",
     "    # the optional dependency of flash-attion is skipped on colab\n",
     "    !pip install wandb louvain\n",
     "\n",
-    "    # NOTE: May need to restart runtime after the installation\n",
+    "    # NOTE: MAY NEED TO RESTART RUNTIME AFTER THE INSTALLATION\n",
     "\n",
     "    print(\"Downloading data and model ckpt...\")\n",
     "    !pip install -q -U gdown\n",

tutorials/zero-shot/Tutorial_ZeroShot_Integration_Continual_Pretraining.ipynb

@@ -46,15 +46,17 @@
     "\n",
     "import os\n",
     "import sys\n",
+    "%load_ext autoreload\n",
+    "%autoreload 2\n",
     "\n",
     "if \"google.colab\" in sys.modules:\n",
     "    print(\"Running on Google Colab\")\n",
     "    print(\"Installing dependencies...\")\n",
-    "    !pip install -U scgpt \"torch<=2.2.2\"\n",
+    "    !pip install -U scgpt \"torch<=2.2.2\" \"numpy<2\" \"umap-learn<0.5.7\"\n",
     "    # the optional dependency of flash-attion is skipped on colab\n",
     "    !pip install wandb louvain\n",
     "\n",
-    "    # NOTE: May need to restart runtime after the installation\n",
+    "    # NOTE: MAY NEED TO RESTART RUNTIME AFTER THE INSTALLATION\n",
     "\n",
     "    print(\"Downloading data and model ckpt...\")\n",
     "    !pip install -q -U gdown\n",

tutorials/zero-shot/Tutorial_ZeroShot_Reference_Mapping.ipynb

@@ -46,15 +46,17 @@
     "\n",
     "import os\n",
     "import sys\n",
+    "%load_ext autoreload\n",
+    "%autoreload 2\n",
     "\n",
     "if \"google.colab\" in sys.modules:\n",
     "    print(\"Running on Google Colab\")\n",
     "    print(\"Installing dependencies...\")\n",
-    "    !pip install -U scgpt \"torch<=2.2.2\"\n",
+    "    !pip install -U scgpt \"torch<=2.2.2\" \"numpy<2\" \"umap-learn<0.5.7\"\n",
     "    # the optional dependency of flash-attion is skipped on colab\n",
     "    !pip install wandb louvain faiss-cpu\n",
     "\n",
-    "    # NOTE: May need to restart runtime after the installation\n",
+    "    # NOTE: MAY NEED TO RESTART RUNTIME AFTER THE INSTALLATION\n",
     "\n",
     "    print(\"Downloading data and model ckpt...\")\n",
     "    !pip install -q -U gdown\n",