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

2023-03-09 22:36:12 +07:00 · 2023-03-09 22:36:12 +07:00 · 0514f011ff
parent 00be48670b
commit 0514f011ff
2 changed files with 3 additions and 3 deletions
--- 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
--- 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)