mrq/tortoise-tts
4
6
Fork 15
Code Issues 9 Pull Requests 1 Packages Projects Releases Wiki Activity

Some fixes

Browse Source
This commit is contained in:
James Betker 2022-03-10 22:56:29 -07:00
parent 8d035595be
commit 54a946d0ae
2 changed files with 5 additions and 3 deletions
Show Stats Download Patch File Download Diff File Expand all files Collapse all files

3
do_tts.py
View File

@ -6,7 +6,7 @@ from urllib import request
import torch import torch
import torch.nn.functional as F import torch.nn.functional as F
import torchaudio import torchaudio
from progressbar import progressbar import progressbar
from models.dvae import DiscreteVAE from models.dvae import DiscreteVAE
from models.autoregressive import UnifiedVoice 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', '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' '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): def show_progress(block_num, block_size, total_size):
global pbar global pbar
if pbar is None: if pbar is None:

3
requirements.txt
View File

@ -3,6 +3,7 @@ torchaudio
rotary_embedding_torch rotary_embedding_torch
transformers transformers
tokenizers tokenizers
pyfastmp3decoder
inflect inflect
progressbar progressbar
einops
unidecode