How to use facebook/musicgen-stereo-small with Transformers:
# Use a pipeline as a high-level helper from transformers import pipeline pipe = pipeline("text-to-audio", model="facebook/musicgen-stereo-small")
# Load model directly from transformers import AutoTokenizer, AutoModelForTextToWaveform tokenizer = AutoTokenizer.from_pretrained("facebook/musicgen-stereo-small") model = AutoModelForTextToWaveform.from_pretrained("facebook/musicgen-stereo-small")
How to use facebook/musicgen-stereo-small with Audiocraft:
from audiocraft.models import MusicGen model = MusicGen.get_pretrained("facebook/musicgen-stereo-small") descriptions = ['happy rock', 'energetic EDM', 'sad jazz'] wav = model.generate(descriptions) # generates 3 samples.
VKlarkV