From 54a946d0ae2753cde33063dc6be7fe1ac1b1101c 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