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

how did I botch this, I don't think it affects anything since it never thrown an error

Browse Source
This commit is contained in:
mrq 2023-03-09 22:36:12 +00:00
parent 00be48670b
commit 0514f011ff
2 changed files with 3 additions and 3 deletions
Show Stats Download Patch File Download Diff File Expand all files Collapse all files

4
requirements.txt
View File

@ -4,10 +4,10 @@ transformers==4.19
tokenizers tokenizers
inflect inflect
progressbar progressbar
einops einops==0.6.0
unidecode unidecode
scipy scipy
librosa librosa==0.8.0
torchaudio torchaudio
threadpoolctl threadpoolctl
appdirs appdirs

2
tortoise/models/autoregressive.py
View File

@ -485,7 +485,7 @@ class UnifiedVoice(nn.Module):
max_generate_length=None, typical_sampling=False, typical_mass=.9, **hf_generate_kwargs): max_generate_length=None, typical_sampling=False, typical_mass=.9, **hf_generate_kwargs):
seq_length = self.max_mel_tokens + self.max_text_tokens + 2 seq_length = self.max_mel_tokens + self.max_text_tokens + 2
if not hasattr(self, 'inference_model'): if not hasattr(self, 'inference_model'):
self.post_init_gpt2_config(kv_cache=self.kv_cachepost_init_gpt2_config) self.post_init_gpt2_config(kv_cache=self.kv_cache)
text_inputs = F.pad(text_inputs, (0, 1), value=self.stop_text_token) text_inputs = F.pad(text_inputs, (0, 1), value=self.stop_text_token)