change quality presets
Browse files- tortoise/api.py +4 -4
tortoise/api.py
CHANGED
@@ -280,10 +280,10 @@ class TextToSpeech:
|
|
280 |
'cond_free_k': 2.0, 'diffusion_temperature': 1.0})
|
281 |
# Presets are defined here.
|
282 |
presets = {
|
283 |
-
'ultra_fast': {'num_autoregressive_samples': 16, 'diffusion_iterations':
|
284 |
-
'fast': {'num_autoregressive_samples': 96, 'diffusion_iterations':
|
285 |
-
'standard': {'num_autoregressive_samples': 256, 'diffusion_iterations':
|
286 |
-
'high_quality': {'num_autoregressive_samples':
|
287 |
}
|
288 |
kwargs.update(presets[preset])
|
289 |
return self.tts(text, **kwargs)
|
|
|
280 |
'cond_free_k': 2.0, 'diffusion_temperature': 1.0})
|
281 |
# Presets are defined here.
|
282 |
presets = {
|
283 |
+
'ultra_fast': {'num_autoregressive_samples': 16, 'diffusion_iterations': 30, 'cond_free': False},
|
284 |
+
'fast': {'num_autoregressive_samples': 96, 'diffusion_iterations': 80},
|
285 |
+
'standard': {'num_autoregressive_samples': 256, 'diffusion_iterations': 200},
|
286 |
+
'high_quality': {'num_autoregressive_samples': 256, 'diffusion_iterations': 400},
|
287 |
}
|
288 |
kwargs.update(presets[preset])
|
289 |
return self.tts(text, **kwargs)
|