From f96d866da3619dc2346b60b124336eee95b1bfb9 Mon Sep 17 00:00:00 2001 From: James Betker Date: Mon, 25 Apr 2022 21:17:49 -0600 Subject: [PATCH] fix notebook --- tortoise_tts.ipynb | 266 ++------------------------------------------- 1 file changed, 11 insertions(+), 255 deletions(-) diff --git a/tortoise_tts.ipynb b/tortoise_tts.ipynb index 0350fda..d8b274f 100644 --- a/tortoise_tts.ipynb +++ b/tortoise_tts.ipynb @@ -34,88 +34,9 @@ "cell_type": "code", "execution_count": null, "metadata": { - "id": "JrK20I32grP6", - "colab": { - "base_uri": "https://localhost:8080/" - }, - "outputId": "44f55dca-5d0a-405e-a4cc-54bc8e16b780" + "id": "JrK20I32grP6" }, - "outputs": [ - { - "output_type": "stream", - "name": "stdout", - "text": [ - "Cloning into 'tortoise-tts'...\n", - "remote: Enumerating objects: 736, done.\u001b[K\n", - "remote: Counting objects: 100% (23/23), done.\u001b[K\n", - "remote: Compressing objects: 100% (15/15), done.\u001b[K\n", - "remote: Total 736 (delta 10), reused 20 (delta 8), pack-reused 713\u001b[K\n", - "Receiving objects: 100% (736/736), 348.62 MiB | 24.08 MiB/s, done.\n", - "Resolving deltas: 100% (161/161), done.\n", - "/content/tortoise-tts\n", - "Requirement already satisfied: torch in /usr/local/lib/python3.7/dist-packages (from -r requirements.txt (line 1)) (1.10.0+cu111)\n", - "Requirement already satisfied: torchaudio in /usr/local/lib/python3.7/dist-packages (from -r requirements.txt (line 2)) (0.10.0+cu111)\n", - "Collecting rotary_embedding_torch\n", - " Downloading rotary_embedding_torch-0.1.5-py3-none-any.whl (4.1 kB)\n", - "Collecting transformers\n", - " Downloading transformers-4.18.0-py3-none-any.whl (4.0 MB)\n", - "\u001b[K |████████████████████████████████| 4.0 MB 5.3 MB/s \n", - "\u001b[?25hCollecting tokenizers\n", - " Downloading tokenizers-0.12.1-cp37-cp37m-manylinux_2_12_x86_64.manylinux2010_x86_64.whl (6.6 MB)\n", - "\u001b[K |████████████████████████████████| 6.6 MB 31.3 MB/s \n", - "\u001b[?25hRequirement already satisfied: inflect in /usr/local/lib/python3.7/dist-packages (from -r requirements.txt (line 6)) (2.1.0)\n", - "Collecting progressbar\n", - " Downloading progressbar-2.5.tar.gz (10 kB)\n", - "Collecting einops\n", - " Downloading einops-0.4.1-py3-none-any.whl (28 kB)\n", - "Collecting unidecode\n", - " Downloading Unidecode-1.3.4-py3-none-any.whl (235 kB)\n", - "\u001b[K |████████████████████████████████| 235 kB 44.3 MB/s \n", - "\u001b[?25hCollecting entmax\n", - " Downloading entmax-1.0.tar.gz (7.2 kB)\n", - "Requirement already satisfied: typing-extensions in /usr/local/lib/python3.7/dist-packages (from torch->-r requirements.txt (line 1)) (4.1.1)\n", - "Requirement already satisfied: tqdm>=4.27 in /usr/local/lib/python3.7/dist-packages (from transformers->-r requirements.txt (line 4)) (4.64.0)\n", - "Requirement already satisfied: packaging>=20.0 in /usr/local/lib/python3.7/dist-packages (from transformers->-r requirements.txt (line 4)) (21.3)\n", - "Collecting sacremoses\n", - " Downloading sacremoses-0.0.49-py3-none-any.whl (895 kB)\n", - "\u001b[K |████████████████████████████████| 895 kB 36.6 MB/s \n", - "\u001b[?25hCollecting huggingface-hub<1.0,>=0.1.0\n", - " Downloading huggingface_hub-0.5.1-py3-none-any.whl (77 kB)\n", - "\u001b[K |████████████████████████████████| 77 kB 6.3 MB/s \n", - "\u001b[?25hRequirement already satisfied: filelock in /usr/local/lib/python3.7/dist-packages (from transformers->-r requirements.txt (line 4)) (3.6.0)\n", - "Collecting pyyaml>=5.1\n", - " Downloading PyYAML-6.0-cp37-cp37m-manylinux_2_5_x86_64.manylinux1_x86_64.manylinux_2_12_x86_64.manylinux2010_x86_64.whl (596 kB)\n", - "\u001b[K |████████████████████████████████| 596 kB 38.9 MB/s \n", - "\u001b[?25hRequirement already satisfied: numpy>=1.17 in /usr/local/lib/python3.7/dist-packages (from transformers->-r requirements.txt (line 4)) (1.21.6)\n", - "Requirement already satisfied: requests in /usr/local/lib/python3.7/dist-packages (from transformers->-r requirements.txt (line 4)) (2.23.0)\n", - "Requirement already satisfied: regex!=2019.12.17 in /usr/local/lib/python3.7/dist-packages (from transformers->-r requirements.txt (line 4)) (2019.12.20)\n", - "Requirement already satisfied: importlib-metadata in /usr/local/lib/python3.7/dist-packages (from transformers->-r requirements.txt (line 4)) (4.11.3)\n", - "Requirement already satisfied: pyparsing!=3.0.5,>=2.0.2 in /usr/local/lib/python3.7/dist-packages (from packaging>=20.0->transformers->-r requirements.txt (line 4)) (3.0.8)\n", - "Requirement already satisfied: zipp>=0.5 in /usr/local/lib/python3.7/dist-packages (from importlib-metadata->transformers->-r requirements.txt (line 4)) (3.8.0)\n", - "Requirement already satisfied: urllib3!=1.25.0,!=1.25.1,<1.26,>=1.21.1 in /usr/local/lib/python3.7/dist-packages (from requests->transformers->-r requirements.txt (line 4)) (1.24.3)\n", - "Requirement already satisfied: chardet<4,>=3.0.2 in /usr/local/lib/python3.7/dist-packages (from requests->transformers->-r requirements.txt (line 4)) (3.0.4)\n", - "Requirement already satisfied: idna<3,>=2.5 in /usr/local/lib/python3.7/dist-packages (from requests->transformers->-r requirements.txt (line 4)) (2.10)\n", - "Requirement already satisfied: certifi>=2017.4.17 in /usr/local/lib/python3.7/dist-packages (from requests->transformers->-r requirements.txt (line 4)) (2021.10.8)\n", - "Requirement already satisfied: six in /usr/local/lib/python3.7/dist-packages (from sacremoses->transformers->-r requirements.txt (line 4)) (1.15.0)\n", - "Requirement already satisfied: joblib in /usr/local/lib/python3.7/dist-packages (from sacremoses->transformers->-r requirements.txt (line 4)) (1.1.0)\n", - "Requirement already satisfied: click in /usr/local/lib/python3.7/dist-packages (from sacremoses->transformers->-r requirements.txt (line 4)) (7.1.2)\n", - "Building wheels for collected packages: progressbar, entmax\n", - " Building wheel for progressbar (setup.py) ... \u001b[?25l\u001b[?25hdone\n", - " Created wheel for progressbar: filename=progressbar-2.5-py3-none-any.whl size=12082 sha256=bb7d90605d0bf4d89aedc46bd8ed39538f55e00ee70fa382c1af81f142f08fa8\n", - " Stored in directory: /root/.cache/pip/wheels/f0/fd/1f/3e35ed57e94cd8ced38dd46771f1f0f94f65fec548659ed855\n", - " Building wheel for entmax (setup.py) ... \u001b[?25l\u001b[?25hdone\n", - " Created wheel for entmax: filename=entmax-1.0-py3-none-any.whl size=11015 sha256=5e2cf723e790ec941984d2030eb3231e1ae3ce75231709391a13edcd2bfb4770\n", - " Stored in directory: /root/.cache/pip/wheels/f7/e8/0d/acc29c2f66e69a1f42483347fa8545c293dec12325ee161716\n", - "Successfully built progressbar entmax\n", - "Installing collected packages: pyyaml, tokenizers, sacremoses, huggingface-hub, einops, unidecode, transformers, rotary-embedding-torch, progressbar, entmax\n", - " Attempting uninstall: pyyaml\n", - " Found existing installation: PyYAML 3.13\n", - " Uninstalling PyYAML-3.13:\n", - " Successfully uninstalled PyYAML-3.13\n", - "Successfully installed einops-0.4.1 entmax-1.0 huggingface-hub-0.5.1 progressbar-2.5 pyyaml-6.0 rotary-embedding-torch-0.1.5 sacremoses-0.0.49 tokenizers-0.12.1 transformers-4.18.0 unidecode-1.3.4\n" - ] - } - ], + "outputs": [], "source": [ "!git clone https://github.com/neonbjb/tortoise-tts.git\n", "%cd tortoise-tts\n", @@ -138,97 +59,10 @@ "tts = TextToSpeech()" ], "metadata": { - "id": "Gen09NM4hONQ", - "colab": { - "base_uri": "https://localhost:8080/" - }, - "outputId": "35c1fb4b-5998-4e75-9ec9-29521b301db6" + "id": "Gen09NM4hONQ" }, "execution_count": null, - "outputs": [ - { - "output_type": "stream", - "name": "stdout", - "text": [ - "Downloading autoregressive.pth from https://huggingface.co/jbetker/tortoise-tts-v2/resolve/hf/.models/autoregressive.pth...\n" - ] - }, - { - "output_type": "stream", - "name": "stderr", - "text": [ - "\n" - ] - }, - { - "output_type": "stream", - "name": "stdout", - "text": [ - "Done.\n", - "Downloading clvp.pth from https://huggingface.co/jbetker/tortoise-tts-v2/resolve/hf/.models/clvp.pth...\n" - ] - }, - { - "output_type": "stream", - "name": "stderr", - "text": [ - "\n" - ] - }, - { - "output_type": "stream", - "name": "stdout", - "text": [ - "Done.\n", - "Downloading cvvp.pth from https://huggingface.co/jbetker/tortoise-tts-v2/resolve/hf/.models/cvvp.pth...\n" - ] - }, - { - "output_type": "stream", - "name": "stderr", - "text": [ - "\n" - ] - }, - { - "output_type": "stream", - "name": "stdout", - "text": [ - "Done.\n", - "Downloading diffusion_decoder.pth from https://huggingface.co/jbetker/tortoise-tts-v2/resolve/hf/.models/diffusion_decoder.pth...\n" - ] - }, - { - "output_type": "stream", - "name": "stderr", - "text": [ - "\n" - ] - }, - { - "output_type": "stream", - "name": "stdout", - "text": [ - "Done.\n", - "Downloading vocoder.pth from https://huggingface.co/jbetker/tortoise-tts-v2/resolve/hf/.models/vocoder.pth...\n" - ] - }, - { - "output_type": "stream", - "name": "stderr", - "text": [ - "\n" - ] - }, - { - "output_type": "stream", - "name": "stdout", - "text": [ - "Done.\n", - "Removing weight norm...\n" - ] - } - ] + "outputs": [] }, { "cell_type": "code", @@ -239,28 +73,10 @@ "%ls voices" ], "metadata": { - "id": "SSleVnRAiEE2", - "colab": { - "base_uri": "https://localhost:8080/" - }, - "outputId": "e1eb09e2-1b68-4f81-b679-edb97538da39" + "id": "SSleVnRAiEE2" }, "execution_count": null, - "outputs": [ - { - "output_type": "stream", - "name": "stdout", - "text": [ - "\u001b[0m\u001b[01;34mangelina_jolie\u001b[0m/ \u001b[01;34mhalle_barry\u001b[0m/ \u001b[01;34mlj\u001b[0m/ \u001b[01;34msamuel_jackson\u001b[0m/\n", - "\u001b[01;34matkins\u001b[0m/ \u001b[01;34mharris\u001b[0m/ \u001b[01;34mmol\u001b[0m/ \u001b[01;34msigourney_weaver\u001b[0m/\n", - "\u001b[01;34mcarlin\u001b[0m/ \u001b[01;34mhenry_cavill\u001b[0m/ \u001b[01;34mmorgan_freeman\u001b[0m/ \u001b[01;34mtom_hanks\u001b[0m/\n", - "\u001b[01;34mdaniel_craig\u001b[0m/ \u001b[01;34mjennifer_lawrence\u001b[0m/ \u001b[01;34mmyself\u001b[0m/ \u001b[01;34mwilliam_shatner\u001b[0m/\n", - "\u001b[01;34mdotrice\u001b[0m/ \u001b[01;34mjohn_krasinski\u001b[0m/ \u001b[01;34motto\u001b[0m/\n", - "\u001b[01;34memma_stone\u001b[0m/ \u001b[01;34mkennard\u001b[0m/ \u001b[01;34mpatrick_stewart\u001b[0m/\n", - "\u001b[01;34mgrace\u001b[0m/ \u001b[01;34mlescault\u001b[0m/ \u001b[01;34mrobert_deniro\u001b[0m/\n" - ] - } - ] + "outputs": [] }, { "cell_type": "code", @@ -302,40 +118,10 @@ "torchaudio.save('generated.wav', gen.squeeze(0).cpu(), 24000)" ], "metadata": { - "id": "KEXOKjIvn6NW", - "colab": { - "base_uri": "https://localhost:8080/" - }, - "outputId": "7977bfd7-9fbc-41f7-d3ac-25fd4e350049" + "id": "KEXOKjIvn6NW" }, "execution_count": null, - "outputs": [ - { - "output_type": "stream", - "name": "stderr", - "text": [ - "100%|██████████| 6/6 [01:18<00:00, 13.11s/it]\n", - "/usr/local/lib/python3.7/dist-packages/torch/utils/checkpoint.py:25: UserWarning: None of the inputs have requires_grad=True. Gradients will be None\n", - " warnings.warn(\"None of the inputs have requires_grad=True. Gradients will be None\")\n", - "/content/tortoise-tts/models/autoregressive.py:359: UserWarning: __floordiv__ is deprecated, and its behavior will change in a future version of pytorch. It currently rounds toward 0 (like the 'trunc' function NOT 'floor'). This results in incorrect rounding for negative values. To keep the current behavior, use torch.div(a, b, rounding_mode='trunc'), or for actual floor division, use torch.div(a, b, rounding_mode='floor').\n", - " mel_lengths = wav_lengths // self.mel_length_compression\n" - ] - }, - { - "output_type": "stream", - "name": "stdout", - "text": [ - "Performing vocoding..\n" - ] - }, - { - "output_type": "stream", - "name": "stderr", - "text": [ - "100%|██████████| 32/32 [00:16<00:00, 1.94it/s]\n" - ] - } - ] + "outputs": [] }, { "cell_type": "code", @@ -346,7 +132,7 @@ "#\n", "# Lets see what it would sound like if Picard and Kirk had a kid with a penchant for philosophy:\n", "conds = []\n", - "for v in ['patrick_stewart', 'william_shatner']:\n", + "for v in ['pat', 'william']:\n", " cond_paths = voices[v]\n", " for cond_path in cond_paths:\n", " c = load_audio(cond_path, 22050)\n", @@ -356,40 +142,10 @@ "torchaudio.save('captain_kirkard.wav', gen.squeeze(0).cpu(), 24000)" ], "metadata": { - "colab": { - "base_uri": "https://localhost:8080/" - }, - "id": "fYTk8KUezUr5", - "outputId": "8a07f251-c90f-4e6a-c204-132b737dfff8" + "id": "fYTk8KUezUr5" }, "execution_count": null, - "outputs": [ - { - "output_type": "stream", - "name": "stderr", - "text": [ - "100%|██████████| 6/6 [01:45<00:00, 17.62s/it]\n", - "/usr/local/lib/python3.7/dist-packages/torch/utils/checkpoint.py:25: UserWarning: None of the inputs have requires_grad=True. Gradients will be None\n", - " warnings.warn(\"None of the inputs have requires_grad=True. Gradients will be None\")\n", - "/content/tortoise-tts/models/autoregressive.py:359: UserWarning: __floordiv__ is deprecated, and its behavior will change in a future version of pytorch. It currently rounds toward 0 (like the 'trunc' function NOT 'floor'). This results in incorrect rounding for negative values. To keep the current behavior, use torch.div(a, b, rounding_mode='trunc'), or for actual floor division, use torch.div(a, b, rounding_mode='floor').\n", - " mel_lengths = wav_lengths // self.mel_length_compression\n" - ] - }, - { - "output_type": "stream", - "name": "stdout", - "text": [ - "Performing vocoding..\n" - ] - }, - { - "output_type": "stream", - "name": "stderr", - "text": [ - "100%|██████████| 32/32 [00:16<00:00, 2.00it/s]\n" - ] - } - ] + "outputs": [] } ] } \ No newline at end of file