neonbjb/tortoise-tts
1
0
Fork 0
mirror of https://github.com/neonbjb/tortoise-tts.git synced 2026-04-21 06:14:04 +00:00
Code Issues Projects Releases Packages Wiki Activity

Merge pull request #564 from rsxdalv/main

Browse source 
strict=False for transformers 4.31.0
This commit is contained in:
manmay nakhashi 2023-08-11 17:31:53 +05:30 • committed by GitHub
commit e96f6e195f
No known key found for this signature in database
GPG key ID: 4AEE18F83AFDEB23
3 changed files with 3 additions and 3 deletions
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,