Passing keyword arguments
In this exercise, you will learn how to pass keyword arguments to model functions when used inside a pipeline.
For this you will be using the MusicGen small model from Meta, which is capable of generating high-quality music samples conditioned on text descriptions or audio prompts.
The pipeline
module has been loaded.
This exercise is part of the course
Multi-Modal Models with Hugging Face
Exercise instructions
- Load a
text-to-audio
pipeline using thefacebook/musicgen-small
model in the PyTorch framework. - Make a dictionary to set the generation temperature to
0.8
andmax_new_tokens
to 2. - Generate an audio array corresponding to the
"Classic rock riff"
prompt.
Hands-on interactive exercise
Have a go at this exercise by completing this sample code.
# Load a text-to-audio pipeline
musicgen = pipeline(task="____", model="____", framework="pt")
# Make a dictionary to set the generation temperature to 0.8
generate_kwargs = ____
# Generate an audio array passing the arguments
outputs = ____
sf.write("output.wav", outputs["audio"][0][0], outputs["sampling_rate"])