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

Updates

Browse source
This commit is contained in:
James Betker 2022-04-12 16:40:42 -06:00
parent 17af2df44f
commit e2ee843098
2 changed files with 4 additions and 4 deletions
Download patch file Download diff file Expand all files Collapse all files

2
api.py
View file

@ -138,7 +138,7 @@ class TextToSpeech:
heads=16, number_text_tokens=256, start_text_token=255, checkpointing=False,
train_solo_embeddings=False,
average_conditioning_embeddings=True).cpu().eval()
self.autoregressive.load_state_dict(torch.load('.models/autoregressive.pth'))
self.autoregressive.load_state_dict(torch.load('.models/autoregressive_diverse.pth'))
self.clip = VoiceCLIP(dim_text=512, dim_speech=512, dim_latent=512, num_text_tokens=256, text_enc_depth=12,
text_seq_len=350, text_heads=8,