From f17128bb13dde4fab657bfabd1b46b1400c609eb Mon Sep 17 00:00:00 2001
From: Vidalnt <raddiscord1@gmail.com>
Date: Sun, 27 Oct 2024 18:38:51 -0500
Subject: [PATCH] fix: :bug: Torch 2.5.0

---
 assets/Applio_NoUI.ipynb | 5 +++++
 1 file changed, 5 insertions(+)

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!\")"
       ]
     },