forked from mrq/tortoise-tts
change quality presets
This commit is contained in:
parent
e23e6f6696
commit
53cb3299d4
|
@ -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)
|
||||||
|
|
Loading…
Reference in New Issue
Block a user