patrickvonplaten
commited on
Commit
•
fd9b508
1
Parent(s):
d0b04f4
Update README.md
Browse files
README.md
CHANGED
@@ -14,10 +14,10 @@ from diffusers import DiffusionPipeline
|
|
14 |
from scipy.io.wavfile import write
|
15 |
|
16 |
model_id = "harmonai/maestro-150k"
|
17 |
-
|
18 |
-
|
19 |
|
20 |
-
audios =
|
21 |
|
22 |
# To save locally
|
23 |
for i, audio in enumerate(audios):
|
@@ -40,8 +40,8 @@ from scipy.io.wavfile import write
|
|
40 |
import torch
|
41 |
|
42 |
model_id = "harmonai/maestro-150k"
|
43 |
-
|
44 |
-
|
45 |
|
46 |
audios = pipeline(audio_length_in_s=4.0).audios
|
47 |
|
|
|
14 |
from scipy.io.wavfile import write
|
15 |
|
16 |
model_id = "harmonai/maestro-150k"
|
17 |
+
pipe = DiffusionPipeline.from_pretrained(model_id)
|
18 |
+
pipe = pipe.to("cuda")
|
19 |
|
20 |
+
audios = pipe(audio_length_in_s=4.0).audios
|
21 |
|
22 |
# To save locally
|
23 |
for i, audio in enumerate(audios):
|
|
|
40 |
import torch
|
41 |
|
42 |
model_id = "harmonai/maestro-150k"
|
43 |
+
pipe = DiffusionPipeline.from_pretrained(model_id, torch_dtype=torch.float16)
|
44 |
+
pipe = pipe.to("cuda")
|
45 |
|
46 |
audios = pipeline(audio_length_in_s=4.0).audios
|
47 |
|