Commit
•
8bdb4cd
1
Parent(s):
5964b12
Update README.md
Browse files
README.md
CHANGED
|
@@ -7,6 +7,7 @@ import scipy
|
|
| 7 |
|
| 8 |
model_id = "harmonai/honk-140k"
|
| 9 |
pipeline = DiffusionPipeline.from_pretrained(model_id)
|
|
|
|
| 10 |
|
| 11 |
audio = pipeline(audio_length_in_s=4.0).audios
|
| 12 |
|
|
@@ -25,6 +26,7 @@ import torch
|
|
| 25 |
|
| 26 |
model_id = "harmonai/honk-140k"
|
| 27 |
pipeline = DiffusionPipeline.from_pretrained(model_id, torch_dtype=torch.float16)
|
|
|
|
| 28 |
|
| 29 |
audio = pipeline(audio_length_in_s=4.0).audios
|
| 30 |
|
|
|
|
| 7 |
|
| 8 |
model_id = "harmonai/honk-140k"
|
| 9 |
pipeline = DiffusionPipeline.from_pretrained(model_id)
|
| 10 |
+
pipeline = pipeline.to("cuda")
|
| 11 |
|
| 12 |
audio = pipeline(audio_length_in_s=4.0).audios
|
| 13 |
|
|
|
|
| 26 |
|
| 27 |
model_id = "harmonai/honk-140k"
|
| 28 |
pipeline = DiffusionPipeline.from_pretrained(model_id, torch_dtype=torch.float16)
|
| 29 |
+
pipeline = pipeline.to("cuda")
|
| 30 |
|
| 31 |
audio = pipeline(audio_length_in_s=4.0).audios
|
| 32 |
|