Spaces:

patriotyk
/

pflowtts_ukr_demo

App Files Files Community

Serhiy Stetskovych commited on Jan 20

Commit

428ef88

•

1 Parent(s): fffaa2a

Move prompt device.

Files changed (1) hide show

app.py +1 -1

app.py CHANGED Viewed

@@ -171,7 +171,7 @@ def synthesise(text, temperature, speed):
         n_timesteps=40,
         temperature=temperature,
         length_scale=1/speed,
-        prompt=normalize(prompt, model.mel_mean, model.mel_std)
     )
     waveform = to_waveform(output["mel"], vocoder, denoiser)

         n_timesteps=40,
         temperature=temperature,
         length_scale=1/speed,
+        prompt=normalize(prompt, model.mel_mean, model.mel_std).to(device)
     )
     waveform = to_waveform(output["mel"], vocoder, denoiser)