CommencerCommencer gratuitement

Transmission d'arguments clés

Dans cet exercice, vous apprendrez comment transmettre des arguments clés à des fonctions de modèle lorsqu'elles sont utilisées dans un pipeline.

Pour ce faire, vous utiliserez le petit modèle MusicGen de Meta, qui est capable de générer des échantillons musicaux à partir de descriptions textuelles ou d'invites audio.

Le module pipeline a été chargé et la bibliothèque soundfile est disponible à l'adresse sf.

Cet exercice fait partie du cours

Modèles multimodaux avec Hugging Face

Afficher le cours

Instructions

  • Chargez un pipeline d'text-to-audio s à l'aide du modèle facebook/musicgen-small dans le framework PyTorch.
  • Créez un dictionnaire pour définir la température de génération sur 0.8 et max_new_tokens sur 1.
  • Générer un tableau audio correspondant à l'invite « "Classic rock riff" ».

Exercice interactif pratique

Essayez cet exercice en complétant cet exemple de 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"])
Modifier et exécuter le code