From 13d466baf529620844c2ce5c0572984119a70b4c Mon Sep 17 00:00:00 2001
From: mrq <barry.quiggles@protonmail.com>
Date: Sat, 18 Feb 2023 16:30:05 +0000
Subject: [PATCH] notebook tweaked, drive mounts and symlinks folders so I can
 stop having to wait a gorillion years to import voices

---
 notebook.ipynb | 27 +++++++++++++++++++++++----
 1 file changed, 23 insertions(+), 4 deletions(-)

diff --git a/notebook.ipynb b/notebook.ipynb
index a2812ba..441936f 100755
--- a/notebook.ipynb
+++ b/notebook.ipynb
@@ -3,7 +3,10 @@
    "nbformat_minor":0,
    "metadata":{
       "colab":{
-         "private_outputs":true
+         "private_outputs":true,
+         "provenance":[
+            
+         ]
       },
       "kernelspec":{
          "name":"python3",
@@ -37,6 +40,7 @@
          "source":[
             "!git clone https://git.ecker.tech/mrq/ai-voice-cloning/\n",
             "%cd ai-voice-cloning\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",
@@ -59,12 +63,14 @@
       {
          "cell_type":"code",
          "source":[
+            "# for my debugging purposes\n",
             "%cd /content/ai-voice-cloning/dlas\n",
             "!git reset --hard HEAD\n",
             "!git pull\n",
             "%cd ..\n",
             "!git reset --hard HEAD\n",
-            "!git pull\n"
+            "!git pull\n",
+            "#exit()"
          ],
          "metadata":{
             "id":"3DktoOXSHmtw"
@@ -86,12 +92,22 @@
       {
          "cell_type":"code",
          "source":[
+            "# only run once, this will save all userdata to your Drive\n",
             "from google.colab import drive\n",
             "drive.mount('/content/drive')\n",
             "\n",
             "%cd /content/ai-voice-cloning\n",
             "!rm -r ./training\n",
-            "!ln -s /content/drive/MyDrive/training/"
+            "!rm -r ./results\n",
+            "!rm -r ./voices\n",
+            "\n",
+            "!mkdir /content/drive/MyDrive/training/\n",
+            "!mkdir /content/drive/MyDrive/results/\n",
+            "!mkdir /content/drive/MyDrive/voices/\n",
+            "\n",
+            "!ln -s /content/drive/MyDrive/training/\n",
+            "!ln -s /content/drive/MyDrive/results/\n",
+            "!ln -s /content/drive/MyDrive/voices/"
          ],
          "metadata":{
             "id":"SGt9gyvubveT"
@@ -134,7 +150,10 @@
             "args = setup_args()\n",
             "\n",
             "webui = setup_gradio()\n",
-            "tts = setup_tortoise()\n",
+            "# Be very, very sure to check \"Defer TTS Load\" in Settings, then restart, before you start training\n",
+            "# You'll crash the runtime if you don't\n",
+            "if not args.defer_tts_load:\n",
+            "\ttts = setup_tortoise()\n",
             "webui.launch(share=True, prevent_thread_lock=True, height=1000)\n",
             "webui.block_thread()"
          ],