Update README.md

This commit is contained in:
Suraj Patil 2022-06-16 18:34:56 +02:00 committed by GitHub
parent 20c722c601
commit 0ffda1dfcc
No known key found for this signature in database
GPG Key ID: 4AEE18F83AFDEB23
1 changed files with 2 additions and 2 deletions

View File

@ -247,11 +247,11 @@ bddm = BDDM.from_pretrained("fusing/diffwave-vocoder-ljspeech")
text = "Hello world, I missed you so much."
# generate mel spectograms using text
mel_spec = grad_tts(text)
mel_spec = grad_tts(text, torch_device=torch_device)
# generate the speech by passing mel spectograms to BDDM pipeline
generator = torch.manual_seed(42)
audio = bddm(mel_spec, generator)
audio = bddm(mel_spec, generator, torch_device=torch_device)
# save generated audio
from scipy.io.wavfile import write as wavwrite