Serhiy Stetskovych commited on
Commit
e42c405
·
1 Parent(s): 1585372

Better params.

Browse files
Files changed (1) hide show
  1. app.py +3 -2
app.py CHANGED
@@ -156,7 +156,7 @@ model = pflowTTS.load_from_checkpoint(PFLOW_MODEL_PATH, map_location=device)
156
  _ = model.eval()
157
  #vocoder = load_bigvgan(VOCODER_BIGVGAN_MODEL_PATH, device)
158
  vocoder = load_hifigan(VOCODER_MODEL_PATH, device)
159
- denoiser = Denoiser(vocoder, mode="zeros")
160
 
161
  @torch.inference_mode()
162
  def synthesise(text, temperature, speed):
@@ -171,7 +171,8 @@ def synthesise(text, temperature, speed):
171
  n_timesteps=40,
172
  temperature=temperature,
173
  length_scale=1/speed,
174
- prompt=normalize(prompt, model.mel_mean, model.mel_std).to(device)
 
175
  )
176
  waveform = to_waveform(output["mel"], vocoder, denoiser)
177
 
 
156
  _ = model.eval()
157
  #vocoder = load_bigvgan(VOCODER_BIGVGAN_MODEL_PATH, device)
158
  vocoder = load_hifigan(VOCODER_MODEL_PATH, device)
159
+ denoiser = None #Denoiser(vocoder, mode="zeros")
160
 
161
  @torch.inference_mode()
162
  def synthesise(text, temperature, speed):
 
171
  n_timesteps=40,
172
  temperature=temperature,
173
  length_scale=1/speed,
174
+ prompt=normalize(prompt, model.mel_mean, model.mel_std).to(device),
175
+ guidance_scale=1.5
176
  )
177
  waveform = to_waveform(output["mel"], vocoder, denoiser)
178