From 76f0f1d453a074df113708eebe12c4bd00f19560 Mon Sep 17 00:00:00 2001 From: patil-suraj Date: Wed, 15 Jun 2022 09:44:18 +0200 Subject: [PATCH] update speech checkpoint name --- README.md | 2 +- 1 file changed, 1 insertion(+), 1 deletion(-) diff --git a/README.md b/README.md index 7c0a2fe71f..210efac201 100644 --- a/README.md +++ b/README.md @@ -197,7 +197,7 @@ from diffusers import BDDM, DiffusionPipeline torch_device = "cuda" # load the BDDM pipeline -bddm = DiffusionPipeline.from_pretrained("fusing/diffwave-vocoder") +bddm = DiffusionPipeline.from_pretrained("fusing/diffwave-vocoder-ljspeech") # load tacotron2 to get the mel spectograms tacotron2 = torch.hub.load('NVIDIA/DeepLearningExamples:torchhub', 'nvidia_tacotron2', model_math='fp16')