Commit
•
ee388db
1
Parent(s):
c35764e
Update README.md
Browse files
README.md
CHANGED
@@ -54,7 +54,7 @@ used to load pre-trained weights and generate text-conditional audio outputs:
|
|
54 |
from diffusers import AudioLDMPipeline
|
55 |
import torch
|
56 |
|
57 |
-
repo_id = "cvssp/audioldm-s-v2"
|
58 |
pipe = AudioLDMPipeline.from_pretrained(repo_id, torch_dtype=torch.float16)
|
59 |
pipe = pipe.to("cuda")
|
60 |
|
|
|
54 |
from diffusers import AudioLDMPipeline
|
55 |
import torch
|
56 |
|
57 |
+
repo_id = "cvssp/audioldm-s-full-v2"
|
58 |
pipe = AudioLDMPipeline.from_pretrained(repo_id, torch_dtype=torch.float16)
|
59 |
pipe = pipe.to("cuda")
|
60 |
|