chaowenguo commited on
Commit
964db7e
·
verified ·
1 Parent(s): 08b8f82

Update README.md

Browse files
Files changed (1) hide show
  1. README.md +2 -2
README.md CHANGED
@@ -7,8 +7,8 @@ inference: false
7
  ```py
8
  import transformers, torch, builtins, numpy
9
 
10
- processor = transformers.AutoProcessor.from_pretrained('facebook/musicgen-stereo-melody-large')
11
- model = transformers.MusicgenMelodyForConditionalGeneration.from_pretrained('facebook/musicgen-stereo-melody-large').to('cuda')
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):