Update README.md

README.md

@@ -26,12 +26,19 @@ By introducing a small delay between the codebooks, we show we can predict them
 
 MusicGen was published in [Simple and Controllable Music Generation](https://arxiv.org/abs/2306.05284) by *Jade Copet, Felix Kreuk, Itai Gat, Tal Remez, David Kant, Gabriel Synnaeve, Yossi Adi, Alexandre Défossez*.
 
-Four checkpoints are released:
-- [**small** (this checkpoint)](https://huggingface.co/facebook/musicgen-small)
-- [medium](https://huggingface.co/facebook/musicgen-medium)
-- [large](https://huggingface.co/facebook/musicgen-large)
-- [melody](https://huggingface.co/facebook/musicgen-melody)
-
+We provide a simple API and 10 pre-trained models. The pre trained models are:
+- `facebook/musicgen-small`: 300M model, text to music only - [:hugging_face: Hub](https://huggingface.co/facebook/musicgen-small)
+- `facebook/musicgen-medium`: 1.5B model, text to music only - [:hugging_face: Hub](https://huggingface.co/facebook/musicgen-medium)
+- `facebook/musicgen-melody`: 1.5B model, text to music and text+melody to music - [:hugging_face: Hub](https://huggingface.co/facebook/musicgen-melody)
+- `facebook/musicgen-large`: 3.3B model, text to music only - [:hugging_face: Hub](https://huggingface.co/facebook/musicgen-large)
+- `facebook/musicgen-melody-large`: 3.3B model, text to music and text+melody to music - [:hugging_face: Hub](https://huggingface.co/facebook/musicgen-melody-large)
+- `facebook/musicgen-stereo-*`: All the previous models fine-tuned for stereo generation -
+    [small](https://huggingface.co/facebook/musicgen-stereo-small),
+    [medium](https://huggingface.co/facebook/musicgen-stereo-medium),
+    [large](https://huggingface.co/facebook/musicgen-stereo-large),
+    [melody](https://huggingface.co/facebook/musicgen-stereo-melody),
+    [melody large](https://huggingface.co/facebook/musicgen-stereo-melody-large)
+  
 ## Example
 
 Try out MusicGen yourself!