reach-vb HF staff commited on
Commit
c424d32
1 Parent(s): d76eb42

Update README.md (#1)

Browse files

- Update README.md (fd228dddc6bd2fe2ef0874377754cd4db2faf2ed)

Files changed (1) hide show
  1. README.md +3 -2
README.md CHANGED
@@ -1,7 +1,8 @@
1
  ---
2
- inference: true
3
  tags:
4
  - magnet
 
5
 
6
  license: cc-by-nc-4.0
7
 
@@ -48,7 +49,7 @@ apt-get install ffmpeg
48
  from audiocraft.models import MAGNeT
49
  from audiocraft.data.audio import audio_write
50
 
51
- model = MAGNeT.get_pretrained("facebook/magnet-small-10secs")
52
 
53
  descriptions = ["happy rock", "energetic EDM"]
54
 
 
1
  ---
2
+ inference: false
3
  tags:
4
  - magnet
5
+ - audiocraft
6
 
7
  license: cc-by-nc-4.0
8
 
 
49
  from audiocraft.models import MAGNeT
50
  from audiocraft.data.audio import audio_write
51
 
52
+ model = MAGNeT.get_pretrained("facebook/audio-magnet-medium")
53
 
54
  descriptions = ["happy rock", "energetic EDM"]
55