Update README.md

README.md

@@ -55,22 +55,23 @@ Try out MusicGen yourself!
 
 ## 🤗 Transformers Usage
 
-You can run MusicGen locally with the 🤗 Transformers library from version 4.31.0 onwards.
+You can run MusicGen Stereo models locally with the 🤗 Transformers library from `main` onward.
 
 1. First install the 🤗 [Transformers library](https://github.com/huggingface/transformers) and scipy:
 
 ```
 pip install --upgrade pip
-pip install --upgrade transformers scipy
+pip install --upgrade git+https://github.com/huggingface/transformers.git scipy
 ```
 
 2. Run inference via the `Text-to-Audio` (TTA) pipeline. You can infer the MusicGen model via the TTA pipeline in just a few lines of code!
 
 ```python
-from transformers import pipeline
 import scipy
+import torch
+from transformers import pipeline
 
-synthesiser = pipeline("text-to-audio", "facebook/musicgen-stereo-large")
+synthesiser = pipeline("text-to-audio", "facebook/musicgen-stereo-large", torch_dtype=torch.float16, device="cuda")
 
 music = synthesiser("lo-fi music with a soothing melody", forward_params={"do_sample": True})
 
@@ -83,13 +84,13 @@ scipy.io.wavfile.write("musicgen_out.wav", rate=music["sampling_rate"], music=au
 from transformers import AutoProcessor, MusicgenForConditionalGeneration
 
 processor = AutoProcessor.from_pretrained("facebook/musicgen-stereo-large")
-model = MusicgenForConditionalGeneration.from_pretrained("facebook/musicgen-stereo-large")
+model = MusicgenForConditionalGeneration.from_pretrained("facebook/musicgen-stereo-large").to("cuda")
 
 inputs = processor(
     text=["80s pop track with bassy drums and synth", "90s rock song with loud guitars and heavy drums"],
     padding=True,
     return_tensors="pt",
-)
+).to("cuda")
 
 audio_values = model.generate(**inputs, max_new_tokens=256)
 ```