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