ПочатиПочніть безкоштовно

Передавання іменованих аргументів

У цій вправі ви навчитеся передавати іменовані аргументи у функції моделі, коли вони використовуються всередині конвеєра (pipeline).

Для цього ви використаєте невелику модель MusicGen від Meta, яка вміє генерувати зразки музики на основі текстових описів або аудіопідказок.

Модуль pipeline вже завантажено, а бібліотека soundfile доступна як sf.

Ця вправа є частиною курсу

Багатомодальні моделі з Hugging Face

Переглянути курс

Інструкції до вправи

  • Завантажте конвеєр text-to-audio, використовуючи модель facebook/musicgen-small на фреймворку PyTorch.
  • Створіть словник generate_kwargs, щоб встановити температуру генерації 0.8 і max_new_tokens на 1.
  • Згенеруйте масив аудіо, використовуючи підказку "Classic rock riff", налаштувавши параметри генерації за допомогою вашого словника generate_kwargs.

Інтерактивна практична вправа

Спробуйте виконати цю вправу, доповнивши цей зразок коду.

# 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"])
Редагувати та запускати код