updated colab notebook

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"

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