Aan de slagGa gratis aan de slag

Keyword-argumenten doorgeven

In deze oefening leer je hoe je keyword-argumenten kunt doorgeven aan modelfuncties wanneer je ze binnen een pipeline gebruikt.

Hiervoor gebruik je het kleine MusicGen-model van Meta, dat muzieksamples kan genereren op basis van tekstbeschrijvingen of audioprompts.

De pipeline-module is geladen en de soundfile-bibliotheek is beschikbaar als sf.

Deze oefening maakt deel uit van de cursus

Multi-modale modellen met Hugging Face

Cursus bekijken

Oefeninstructies

  • Laad een text-to-audio-pipeline met het facebook/musicgen-small-model in het PyTorch-framework.
  • Maak een dictionary met de naam generate_kwargs om de generatietemperatuur op 0.8 te zetten en max_new_tokens op 1.
  • Genereer een audio-array met de prompt "Classic rock riff" en pas de generatieparameters aan met je generate_kwargs-dictionary.

Praktische interactieve oefening

Probeer deze oefening eens door deze voorbeeldcode in te vullen.

# 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"])
Code bewerken en uitvoeren