File size: 472 Bytes
31c1c5c | 1 2 3 4 5 6 7 8 9 10 11 12 13 14 15 | from transformers import PretrainedConfig
from huggingface_integration import AudioCraftForHuggingFace
# Example usage:
model_name = 'facebook/audiogen-medium'
duration = 5
config = PretrainedConfig.from_pretrained(model_name)
audio_craft_model = AudioCraftForHuggingFace(config, model_name, duration)
descriptions = ['creepy cartoon dungeon music']
wav = audio_craft_model(descriptions)
for idx, one_wav in enumerate(wav):
audio_craft_model.save_wav(one_wav, idx) |