@@ -25,14 +25,14 @@ This pipeline was contributed by [sanchit-gandhi](https://huggingface.co/sanchit
2525
2626## Text-to-Audio
2727
28- The [`AudioLDMPipeline`] can be used to load pre-trained weights from [cvssp/audioldm](https://huggingface.co/cvssp/audioldm) and generate text-conditional audio outputs:
28+ The [`AudioLDMPipeline`] can be used to load pre-trained weights from [cvssp/audioldm-s-full-v2 ](https://huggingface.co/cvssp/audioldm-s-full-v2 ) and generate text-conditional audio outputs:
2929
3030```python
3131from diffusers import AudioLDMPipeline
3232import torch
3333import scipy
3434
35- repo_id = " cvssp/audioldm"
35+ repo_id = " cvssp/audioldm-s-full-v2 "
3636pipe = AudioLDMPipeline.from_pretrained(repo_id, torch_dtype =torch.float16)
3737pipe = pipe.to("cuda")
3838
@@ -56,7 +56,7 @@ Inference:
5656### How to load and use different schedulers
5757
5858The AudioLDM pipeline uses [`DDIMScheduler`] scheduler by default. But `diffusers` provides many other schedulers
59- that can be used with the AudioLDM pipeline such as [`PNDMScheduler`], [`LMSDiscreteScheduler`], [`EulerDiscreteScheduler`],
59+ that can be used with the AudioLDM pipeline such as [`PNDMScheduler`], [`LMSDiscreteScheduler`], [`EulerDiscreteScheduler`],
6060[`EulerAncestralDiscreteScheduler`] etc. We recommend using the [`DPMSolverMultistepScheduler`] as it's currently the fastest
6161scheduler there is.
6262
@@ -68,12 +68,14 @@ method, or pass the `scheduler` argument to the `from_pretrained` method of the
6868>>> from diffusers import AudioLDMPipeline, DPMSolverMultistepScheduler
6969>>> import torch
7070
71- >>> pipeline = AudioLDMPipeline .from_pretrained (" cvssp/audioldm" , torch_dtype =torch .float16 )
71+ >>> pipeline = AudioLDMPipeline .from_pretrained (" cvssp/audioldm-s-full-v2 " , torch_dtype =torch .float16 )
7272>>> pipeline .scheduler = DPMSolverMultistepScheduler .from_config (pipeline .scheduler .config )
7373
7474>>> # or
75- >>> dpm_scheduler = DPMSolverMultistepScheduler .from_pretrained (" cvssp/audioldm" , subfolder =" scheduler" )
76- >>> pipeline = AudioLDMPipeline .from_pretrained (" cvssp/audioldm" , scheduler =dpm_scheduler , torch_dtype =torch .float16 )
75+ >>> dpm_scheduler = DPMSolverMultistepScheduler .from_pretrained (" cvssp/audioldm-s-full-v2" , subfolder =" scheduler" )
76+ >>> pipeline = AudioLDMPipeline .from_pretrained (
77+ ... " cvssp/audioldm-s-full-v2" , scheduler =dpm_scheduler , torch_dtype =torch .float16
78+ ... )
7779` ` `
7880
7981## AudioLDMPipeline
0 commit comments