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 music samples based on text descriptions or audio prompts.
The pipeline
module has been loaded, and the soundfile
library is available as sf
.
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
to1
. - 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 and max_new_tokens to 1
generate_kwargs = {"____": ____, "____": ____}
# Generate an audio array passing the arguments
outputs = ____
sf.write("output.wav", outputs["audio"][0][0], outputs["sampling_rate"])