update readme aef8e3f
sanchit-gandhi commited on
How to use facebook/musicgen-medium with Transformers:
# Use a pipeline as a high-level helper
from transformers import pipeline
pipe = pipeline("text-to-audio", model="facebook/musicgen-medium") # Load model directly
from transformers import AutoProcessor, AutoModelForTextToWaveform
processor = AutoProcessor.from_pretrained("facebook/musicgen-medium")
model = AutoModelForTextToWaveform.from_pretrained("facebook/musicgen-medium", device_map="auto")