From 50cd14fd3d4dd04304dc572325605b35bbf67abe Mon Sep 17 00:00:00 2001 From: Roberts Slisans Date: Fri, 11 Aug 2023 14:58:43 +0300 Subject: [PATCH] strict=False for transformers 4.31.0 --- requirements.txt | 2 +- setup.py | 2 +- tortoise/api.py | 2 +- 3 files changed, 3 insertions(+), 3 deletions(-) diff --git a/requirements.txt b/requirements.txt index 5168c32..d5e7cb2 100644 --- a/requirements.txt +++ b/requirements.txt @@ -1,6 +1,6 @@ tqdm rotary_embedding_torch -transformers==4.29.2 +transformers==4.31.0 tokenizers inflect progressbar diff --git a/setup.py b/setup.py index 2ca4aed..45cbe30 100644 --- a/setup.py +++ b/setup.py @@ -27,7 +27,7 @@ setuptools.setup( 'unidecode', 'scipy', 'librosa', - 'transformers==4.29.2', + 'transformers==4.31.0', 'tokenizers', ], classifiers=[ diff --git a/tortoise/api.py b/tortoise/api.py index efa01fb..cc76db8 100644 --- a/tortoise/api.py +++ b/tortoise/api.py @@ -227,7 +227,7 @@ class TextToSpeech: model_dim=1024, heads=16, number_text_tokens=255, start_text_token=255, checkpointing=False, train_solo_embeddings=False).cpu().eval() - self.autoregressive.load_state_dict(torch.load(get_model_path('autoregressive.pth', models_dir))) + self.autoregressive.load_state_dict(torch.load(get_model_path('autoregressive.pth', models_dir)), strict=False) self.autoregressive.post_init_gpt2_config(use_deepspeed=use_deepspeed, kv_cache=kv_cache, half=self.half) self.diffusion = DiffusionTts(model_channels=1024, num_layers=10, in_channels=100, out_channels=200,