Commit
•
b8135e3
1
Parent(s):
793d2ab
Update README.md
Browse files
README.md
CHANGED
@@ -47,7 +47,7 @@ used to load pre-trained weights and generate text-conditional audio outputs:
|
|
47 |
from diffusers import MusicLDMPipeline
|
48 |
import torch
|
49 |
|
50 |
-
repo_id = "
|
51 |
pipe = MusicLDMPipeline.from_pretrained(repo_id, torch_dtype=torch.float16)
|
52 |
pipe = pipe.to("cuda")
|
53 |
|
@@ -90,7 +90,7 @@ import torch
|
|
90 |
from diffusers import MusicLDMPipeline
|
91 |
|
92 |
# load the pipeline
|
93 |
-
repo_id = "
|
94 |
pipe = MusicLDMPipeline.from_pretrained(repo_id, torch_dtype=torch.float16)
|
95 |
pipe = pipe.to("cuda")
|
96 |
|
|
|
47 |
from diffusers import MusicLDMPipeline
|
48 |
import torch
|
49 |
|
50 |
+
repo_id = "ucsd-reach/musicldm"
|
51 |
pipe = MusicLDMPipeline.from_pretrained(repo_id, torch_dtype=torch.float16)
|
52 |
pipe = pipe.to("cuda")
|
53 |
|
|
|
90 |
from diffusers import MusicLDMPipeline
|
91 |
|
92 |
# load the pipeline
|
93 |
+
repo_id = "ucsd-reach/musicldm"
|
94 |
pipe = MusicLDMPipeline.from_pretrained(repo_id, torch_dtype=torch.float16)
|
95 |
pipe = pipe.to("cuda")
|
96 |
|