fix: 🐛 Torch 2.5.0

IAHispano · Oct 27, 2024 · f17128b · f17128b
1 parent 7bc069f
commit f17128b
Showing 1 changed file with 5 additions and 0 deletions.
diff --git 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!\")"
       ]
     },