From 94f88886b03a2f1548677ce6139c84ecd39221a3 Mon Sep 17 00:00:00 2001
From: mrq <mrq@ecker.tech>
Date: Tue, 24 Oct 2023 23:39:46 +0000
Subject: [PATCH] updated colab notebook

---
 notebook_colab.ipynb | 21 ++++++++++++++++++---
 requirements.txt     |  6 +++++-
 2 files changed, 23 insertions(+), 4 deletions(-)

diff --git a/notebook_colab.ipynb b/notebook_colab.ipynb
index 74de2ba..744ba56 100755
--- a/notebook_colab.ipynb
+++ b/notebook_colab.ipynb
@@ -38,10 +38,24 @@
             
          ],
          "source":[
-            "!apt install python3.8-venv\n",
+            "!apt install python3.10-venv\n",
             "!git clone https://git.ecker.tech/mrq/ai-voice-cloning/\n",
             "%cd /content/ai-voice-cloning\n",
-            "!./setup-cuda.sh"
+            "# get local dependencies\n",
+            "!git submodule init\n",
+            "!git submodule update --remote\n",
+            "# setup venv\n",
+            "!python3 -m venv venv\n",
+            "!source ./venv/bin/activate\n",
+            "!python3 -m pip install --upgrade pip # just to be safe\n",
+            "# CUDA\n",
+            "!pip3 install torch torchvision torchaudio --index-url https://download.pytorch.org/whl/cu118\n",
+            "# install requirements\n",
+            "!python3 -m pip install -r ./modules/tortoise-tts/requirements.txt # install TorToiSe requirements\n",
+            "!python3 -m pip install -e ./modules/tortoise-tts/ # install TorToiSe\n",
+            "!python3 -m pip install -r ./modules/dlas/requirements.txt # instal DLAS requirements, last, because whisperx will break a dependency here\n",
+            "!python3 -m pip install -e ./modules/dlas/ # install DLAS\n",
+            "!python3 -m pip install -r ./requirements.txt # install local requirements"
          ]
       },
       {
@@ -115,7 +129,8 @@
          "cell_type":"code",
          "source":[
             "%cd /content/ai-voice-cloning/\n",
-            "!./start.sh --share"
+            "!source ./venv/bin/activate\n",
+            "!python3 ./src/main.py --share"
          ],
          "metadata":{
             "id":"QRA8jF3cF-YJ"
diff --git a/requirements.txt b/requirements.txt
index d062a19..df9fe9e 100755
--- a/requirements.txt
+++ b/requirements.txt
@@ -1,5 +1,9 @@
-git+https://github.com/openai/whisper.git
+--extra-index-url https://download.pytorch.org/whl/cu118
+torch>=2.1.0
+torchvision
+torchaudio
 
+openai-whisper
 more-itertools
 ffmpeg-python
 gradio<=3.23.0