From 53cb3299d4c775dd502aded3485a36b39479ff8e Mon Sep 17 00:00:00 2001 From: James Betker Date: Tue, 3 May 2022 21:01:26 -0600 Subject: [PATCH] change quality presets --- tortoise/api.py | 8 ++++---- 1 file changed, 4 insertions(+), 4 deletions(-) diff --git a/tortoise/api.py b/tortoise/api.py index c4cce4d..11a13ff 100644 --- a/tortoise/api.py +++ b/tortoise/api.py @@ -280,10 +280,10 @@ class TextToSpeech: 'cond_free_k': 2.0, 'diffusion_temperature': 1.0}) # Presets are defined here. presets = { - 'ultra_fast': {'num_autoregressive_samples': 16, 'diffusion_iterations': 32, 'cond_free': False}, - 'fast': {'num_autoregressive_samples': 96, 'diffusion_iterations': 32}, - 'standard': {'num_autoregressive_samples': 256, 'diffusion_iterations': 128}, - 'high_quality': {'num_autoregressive_samples': 512, 'diffusion_iterations': 1024}, + 'ultra_fast': {'num_autoregressive_samples': 16, 'diffusion_iterations': 30, 'cond_free': False}, + 'fast': {'num_autoregressive_samples': 96, 'diffusion_iterations': 80}, + 'standard': {'num_autoregressive_samples': 256, 'diffusion_iterations': 200}, + 'high_quality': {'num_autoregressive_samples': 256, 'diffusion_iterations': 400}, } kwargs.update(presets[preset]) return self.tts(text, **kwargs)