neonbjb/tortoise-tts
1
0
Fork 0
mirror of https://github.com/neonbjb/tortoise-tts.git synced 2026-03-10 23:43:50 +01:00
Code Issues Actions Packages Projects Releases Wiki Activity

strict=False for transformers 4.31.0

Browse source
This commit is contained in:
Roberts Slisans 2023-08-11 14:58:43 +03:00 committed by GitHub
parent 3c4d9c5131
commit 50cd14fd3d
No known key found for this signature in database
GPG key ID: 4AEE18F83AFDEB23
3 changed files with 3 additions and 3 deletions
Show stats Download patch file Download diff file Expand all files Collapse all files

2
requirements.txt
View file

@ -1,6 +1,6 @@
tqdm
rotary_embedding_torch
transformers==4.29.2
transformers==4.31.0
tokenizers
inflect
progressbar

2
setup.py
View file

@ -27,7 +27,7 @@ setuptools.setup(
'unidecode',
'scipy',
'librosa',
'transformers==4.29.2',
'transformers==4.31.0',
'tokenizers',
],
classifiers=[

2
tortoise/api.py
View file

@ -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,