From 0514f011ff22e7864a5b0976ca5c0c0545dc42f8 Mon Sep 17 00:00:00 2001 From: mrq Date: Thu, 9 Mar 2023 22:36:12 +0000 Subject: [PATCH] how did I botch this, I don't think it affects anything since it never thrown an error --- requirements.txt | 4 ++-- tortoise/models/autoregressive.py | 2 +- 2 files changed, 3 insertions(+), 3 deletions(-) diff --git a/requirements.txt b/requirements.txt index 66270da..594507f 100755 --- a/requirements.txt +++ b/requirements.txt @@ -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 diff --git a/tortoise/models/autoregressive.py b/tortoise/models/autoregressive.py index b07a3b3..1207a5f 100755 --- a/tortoise/models/autoregressive.py +++ b/tortoise/models/autoregressive.py @@ -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)