Add transformers weights & configs f0c61e0
Sanchit Gandhi commited on
How to use facebook/musicgen-stereo-medium with Transformers:
# Use a pipeline as a high-level helper
from transformers import pipeline
pipe = pipeline("text-to-audio", model="facebook/musicgen-stereo-medium") # Load model directly
from transformers import AutoProcessor, AutoModelForTextToWaveform
processor = AutoProcessor.from_pretrained("facebook/musicgen-stereo-medium")
model = AutoModelForTextToWaveform.from_pretrained("facebook/musicgen-stereo-medium", device_map="auto")How to use facebook/musicgen-stereo-medium with Audiocraft:
from audiocraft.models import MusicGen
model = MusicGen.get_pretrained("facebook/musicgen-stereo-medium")
descriptions = ['happy rock', 'energetic EDM', 'sad jazz']
wav = model.generate(descriptions) # generates 3 samples.