Update README.md

This commit is contained in:
Patrick von Platen 2022-06-22 18:38:32 +02:00 committed by GitHub
parent 3a17775454
commit 6e456b7a7a
No known key found for this signature in database
GPG Key ID: 4AEE18F83AFDEB23
1 changed files with 2 additions and 2 deletions

View File

@ -253,12 +253,12 @@ image_pil.save("test.png")
```python
import torch
from diffusers import BDDMPipeline, GradTTS
from diffusers import BDDMPipeline, GradTTSPipeline
torch_device = "cuda"
# load grad tts and bddm pipelines
grad_tts = GradTTS.from_pretrained("fusing/grad-tts-libri-tts")
grad_tts = GradTTSPipeline.from_pretrained("fusing/grad-tts-libri-tts")
bddm = BDDMPipeline.from_pretrained("fusing/diffwave-vocoder-ljspeech")
text = "Hello world, I missed you so much."