Update README.md
Browse files
README.md
CHANGED
@@ -7,8 +7,8 @@ inference: false
|
|
7 |
```py
|
8 |
import transformers, torch, builtins, numpy
|
9 |
|
10 |
-
processor = transformers.AutoProcessor.from_pretrained('
|
11 |
-
model = transformers.MusicgenMelodyForConditionalGeneration.from_pretrained('
|
12 |
|
13 |
result = []
|
14 |
for _ in builtins.range(9):
|
|
|
7 |
```py
|
8 |
import transformers, torch, builtins, numpy
|
9 |
|
10 |
+
processor = transformers.AutoProcessor.from_pretrained('chaowenguo/musicgen')
|
11 |
+
model = transformers.MusicgenMelodyForConditionalGeneration.from_pretrained('chaowenguo/musicgen').to('cuda')
|
12 |
|
13 |
result = []
|
14 |
for _ in builtins.range(9):
|