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 |
|