diff --git a/notebooks/comfyui_colab_with_manager.ipynb b/notebooks/comfyui_colab_with_manager.ipynb
index 6dc8080b..cf4f9a56 100644
--- a/notebooks/comfyui_colab_with_manager.ipynb
+++ b/notebooks/comfyui_colab_with_manager.ipynb
@@ -58,6 +58,7 @@
         "!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",
+        "!pip3 install torchsde\n",
         "\n",
         "if OPTIONS['USE_COMFYUI_MANAGER']:\n",
         "  %cd custom_nodes\n",