Passando argumentos nomeados
Neste exercício, você vai aprender como passar argumentos nomeados para funções de modelo quando usadas dentro de um pipeline.
Para isso, você usará o modelo MusicGen small da Meta, capaz de gerar trechos de música a partir de descrições em texto ou prompts de áudio.
O módulo pipeline já foi carregado e a biblioteca soundfile está disponível como sf.
Este exercício faz parte do curso
Modelos multimodais com Hugging Face
Instruções do exercício
- Carregue um pipeline de
text-to-audiousando o modelofacebook/musicgen-smallno framework PyTorch. - Crie um dicionário chamado
generate_kwargspara definir a temperatura de geração como0.8emax_new_tokenscomo1. - Gere um array de áudio usando o prompt
"Classic rock riff", ajustando os parâmetros de geração com o seu dicionáriogenerate_kwargs.
Exercício interativo prático
Experimente este exercício completando este código de exemplo.
# 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"])