diff --git a/notebooks/comfyui_colab_with_manager.ipynb b/notebooks/comfyui_colab_with_manager.ipynb
index 1b00e5a..6dc8080 100644
--- a/notebooks/comfyui_colab_with_manager.ipynb
+++ b/notebooks/comfyui_colab_with_manager.ipynb
@@ -55,7 +55,9 @@
         "!echo -= Install dependencies =-\n",
         "#Remove cu121 as it causes issues in Colab.\n",
         "#!pip install xformers!=0.0.18 -r requirements.txt --extra-index-url https://download.pytorch.org/whl/cu121 --extra-index-url https://download.pytorch.org/whl/cu118 --extra-index-url https://download.pytorch.org/whl/cu117\n",
-        "!pip3 install torch torchsde einops transformers>=4.25.1 safetensors>=0.3.0 aiohttp accelerate pyyaml Pillow scipy tqdm psutil --index-url https://download.pytorch.org/whl/cu118\n",
+        "!pip3 install accelerate\n",
+        "!pip3 install einops transformers>=4.25.1 safetensors>=0.3.0 aiohttp pyyaml Pillow scipy tqdm psutil\n",
+        "!pip3 install xformers!=0.0.18 torch==2.1.0 torchvision torchaudio --index-url https://download.pytorch.org/whl/cu118\n",
         "\n",
         "if OPTIONS['USE_COMFYUI_MANAGER']:\n",
         "  %cd custom_nodes\n",