Jarod/tortoise-tts
1
1
Fork 0
forked from mrq/tortoise-tts
Code Pull Requests Activity

change quality presets

Browse Source
This commit is contained in:
James Betker 2022-05-03 21:01:26 -06:00
parent e23e6f6696
commit 53cb3299d4
1 changed files with 4 additions and 4 deletions
Show Stats Download Patch File Download Diff File Expand all files Collapse all files

8
tortoise/api.py
View File

@ -280,10 +280,10 @@ class TextToSpeech:
'cond_free_k': 2.0, 'diffusion_temperature': 1.0}) 'cond_free_k': 2.0, 'diffusion_temperature': 1.0})
# Presets are defined here. # Presets are defined here.
presets = { presets = {
'ultra_fast': {'num_autoregressive_samples': 16, 'diffusion_iterations': 32, 'cond_free': False}, 'ultra_fast': {'num_autoregressive_samples': 16, 'diffusion_iterations': 30, 'cond_free': False},
'fast': {'num_autoregressive_samples': 96, 'diffusion_iterations': 32}, 'fast': {'num_autoregressive_samples': 96, 'diffusion_iterations': 80},
'standard': {'num_autoregressive_samples': 256, 'diffusion_iterations': 128}, 'standard': {'num_autoregressive_samples': 256, 'diffusion_iterations': 200},
'high_quality': {'num_autoregressive_samples': 512, 'diffusion_iterations': 1024}, 'high_quality': {'num_autoregressive_samples': 256, 'diffusion_iterations': 400},
} }
kwargs.update(presets[preset]) kwargs.update(presets[preset])
return self.tts(text, **kwargs) return self.tts(text, **kwargs)