jbetker commited on
Commit
c1d004a
1 Parent(s): ee6f9b1

change quality presets

Browse files
Files changed (1) hide show
  1. 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': 32, 'cond_free': False},
284
- 'fast': {'num_autoregressive_samples': 96, 'diffusion_iterations': 32},
285
- 'standard': {'num_autoregressive_samples': 256, 'diffusion_iterations': 128},
286
- 'high_quality': {'num_autoregressive_samples': 512, 'diffusion_iterations': 1024},
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)