diff --git a/assets/Applio_NoUI.ipynb b/assets/Applio_NoUI.ipynb
index 1105dc13..57e2a76c 100644
--- a/assets/Applio_NoUI.ipynb
+++ b/assets/Applio_NoUI.ipynb
@@ -107,6 +107,8 @@
         "import tarfile\n",
         "import subprocess\n",
         "from pathlib import Path\n",
+        "from IPython.display import clear_output\n",
+        "\n",
         "def vidal_setup(C):\n",
         "    def F():\n",
         "        print(\"Installing pip packages...\")\n",
@@ -143,6 +145,9 @@
         "\n",
         "\n",
         "vidal_setup(False)\n",
+        "!pip uninstall torch torchvision torchaudio -y\n",
+        "!pip install torch==2.3.1 torchvision==0.18.1 torchaudio==2.3.1 --upgrade --index-url https://download.pytorch.org/whl/cu121\n",
+        "clear_output()\n",
         "print(\"Finished installing requirements!\")"
       ]
     },