From ad5cbffc718f664927216fc1de3dc9dc50b7d084 Mon Sep 17 00:00:00 2001
From: James Betker <jbetker@gmail.com>
Date: Thu, 10 Mar 2022 22:56:29 -0700
Subject: [PATCH] Some fixes
---
do_tts.py | 3 ++-
requirements.txt | 5 +++--
2 files changed, 5 insertions(+), 3 deletions(-)
diff --git a/do_tts.py b/do_tts.py
index e813261..a8951d0 100644
--- a/do_tts.py
+++ b/do_tts.py
@@ -6,7 +6,7 @@ from urllib import request
import torch
import torch.nn.functional as F
import torchaudio
-from progressbar import progressbar
+import progressbar
from models.dvae import DiscreteVAE
from models.autoregressive import UnifiedVoice
@@ -27,6 +27,7 @@ def download_models():
'diffusion.pth': 'https://huggingface.co/jbetker/tortoise-tts-diffusion-v1/resolve/main/pytorch-model.bin',
'autoregressive.pth': 'https://huggingface.co/jbetker/tortoise-tts-autoregressive/resolve/main/pytorch-model.bin'
}
+ os.makedirs('.models', exist_ok=True)
def show_progress(block_num, block_size, total_size):
global pbar
if pbar is None:
diff --git a/requirements.txt b/requirements.txt
index ddddd85..880c033 100644
--- a/requirements.txt
+++ b/requirements.txt
@@ -3,6 +3,7 @@ torchaudio
rotary_embedding_torch
transformers
tokenizers
-pyfastmp3decoder
inflect
-progressbar
\ No newline at end of file
+progressbar
+einops
+unidecode
\ No newline at end of file