From d58b67004af54923caa4f891101e4f8988e1b774 Mon Sep 17 00:00:00 2001
From: mrq <barry.quiggles@protonmail.com>
Date: Wed, 8 Mar 2023 15:51:13 +0000
Subject: [PATCH] colab notebook uses venv and normal scripts to keep it on
 parity with a local install (and it literally just works stop creating issues
 for someething inconsistent with known solutions)

---
 notebook_colab.ipynb | 75 +++-----------------------------------------
 1 file changed, 4 insertions(+), 71 deletions(-)

diff --git a/notebook_colab.ipynb b/notebook_colab.ipynb
index 381648e..372fd53 100755
--- a/notebook_colab.ipynb
+++ b/notebook_colab.ipynb
@@ -38,25 +38,10 @@
             
          ],
          "source":[
+            "!apt install python3.8-venv\n",
             "!git clone https://git.ecker.tech/mrq/ai-voice-cloning/\n",
-            "%cd ai-voice-cloning\n",
-            "\n",
-            "!git submodule init\n",
-            "!git submodule update --remote\n",
-            "\n",
-            "# TODO: fix venvs working for subprocess.Popen calling a bash script\n",
-            "#!apt install python3.8-venv\n",
-            "#!python -m venv venv\n",
-            "#!source ./venv/bin/activate\n",
-            "\n",
-            "!python -m pip install --upgrade pip\n",
-            "!pip install torch torchvision torchaudio --extra-index-url https://download.pytorch.org/whl/cu116\n",
-            "!python -m pip install -r ./requirements.txt\n",
-            "!python -m pip install -r ./tortoise-tts/requirements.txt\n",
-            "!python -m pip install -e ./tortoise-tts/\n",
-            "!python -m pip install -r ./dlas/requirements.txt\n",
-            "\n",
-            "!rm ./tortoise-tts/{main,webui}.py"
+            "%cd /content/ai-voice-cloning\n",
+            "!./setup-cuda.sh"
          ]
       },
       {
@@ -120,58 +105,7 @@
       {
          "cell_type":"markdown",
          "source":[
-            "## Running (Inlined)"
-         ],
-         "metadata":{
-            "id":"o1gkfw3B3JSk"
-         }
-      },
-      {
-         "cell_type":"code",
-         "source":[
-            "%cd /content/ai-voice-cloning\n",
-            "#!source ./venv/bin/activate\n",
-            "\n",
-            "import os\n",
-            "import sys\n",
-            "\n",
-            "sys.argv = [\"\"]\n",
-            "if './src/' not in sys.path:\n",
-            "\tsys.path.append('./src/')\n",
-            "if './tortoise-tts/' not in sys.path:\n",
-            "\tsys.path.append('./tortoise-tts/')\n",
-            "\n",
-            "if 'TORTOISE_MODELS_DIR' not in os.environ:\n",
-            "\tos.environ['TORTOISE_MODELS_DIR'] = os.path.realpath(os.path.join(os.getcwd(), './models/tortoise/'))\n",
-            "\n",
-            "if 'TRANSFORMERS_CACHE' not in os.environ:\n",
-            "\tos.environ['TRANSFORMERS_CACHE'] = os.path.realpath(os.path.join(os.getcwd(), './models/transformers/'))\n",
-            "\n",
-            "import utils\n",
-            "import webui\n",
-            "\n",
-            "args = utils.setup_args()\n",
-            "ui = webui.setup_gradio()\n",
-            "# Be very, very sure to check \"Do Not Load TTS On Startup\" in Settings after all the models download, then restart, before you start training\n",
-            "# You'll crash the runtime if you don't\n",
-            "if not args.defer_tts_load:\n",
-            "\tutils.setup_tortoise()\n",
-            "\n",
-            "ui.launch(share=True, prevent_thread_lock=True, height=1000)\n",
-            "ui.block_thread()"
-         ],
-         "metadata":{
-            "id":"c_EQZLTA19c7"
-         },
-         "execution_count":null,
-         "outputs":[
-            
-         ]
-      },
-      {
-         "cell_type":"markdown",
-         "source":[
-            "## Running (non-inlined)"
+            "## Running"
          ],
          "metadata":{
             "id":"EM3iNqgJF6Be"
@@ -203,7 +137,6 @@
       {
          "cell_type":"code",
          "source":[
-            "!rm ./tortoise-tts/{main,webui}.py\n",
             "exit()"
          ],
          "metadata":{