Get startedGet started for free

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

View Course

Exercise instructions

  • Load a text-to-audio pipeline using the facebook/musicgen-small model in the PyTorch framework.
  • Make a dictionary to set the generation temperature to 0.8 and max_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"])
Edit and Run Code