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

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

View File

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

View File

@ -485,7 +485,7 @@ class UnifiedVoice(nn.Module):
max_generate_length=None, typical_sampling=False, typical_mass=.9, **hf_generate_kwargs):
seq_length = self.max_mel_tokens + self.max_text_tokens + 2
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)