Update README.md

README.md

@@ -75,10 +75,11 @@ 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-small", device="cuda")
+synthesiser = pipeline("text-to-audio", "facebook/musicgen-stereo-small", device="cuda", torch_dtype=torch.float16)
 
 music = synthesiser("lo-fi music with a soothing melody", forward_params={"do_sample": True})