Update README.md
Browse files
README.md
CHANGED
@@ -78,8 +78,8 @@ pip install --upgrade git+https://github.com/huggingface/transformers.git scipy
|
|
78 |
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!
|
79 |
|
80 |
```python
|
81 |
-
import soundfile
|
82 |
import torch
|
|
|
83 |
from transformers import pipeline
|
84 |
|
85 |
synthesiser = pipeline("text-to-audio", "facebook/musicgen-stereo-small", device="cuda:0", torch_dtype=torch.float16)
|
|
|
78 |
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!
|
79 |
|
80 |
```python
|
|
|
81 |
import torch
|
82 |
+
import soundfile as sf
|
83 |
from transformers import pipeline
|
84 |
|
85 |
synthesiser = pipeline("text-to-audio", "facebook/musicgen-stereo-small", device="cuda:0", torch_dtype=torch.float16)
|