From 95fc40225cdacbe1bd69f1ef75b44dba66ed3ae5 Mon Sep 17 00:00:00 2001 From: Igor Skovorodkin Date: Mon, 11 Nov 2024 21:09:45 +0300 Subject: [PATCH] Update MUSICGEN_STYLE.md text-and-style conditioning generation sample fix --- docs/MUSICGEN_STYLE.md | 2 +- 1 file changed, 1 insertion(+), 1 deletion(-) diff --git a/docs/MUSICGEN_STYLE.md b/docs/MUSICGEN_STYLE.md index 9962b2ea..8c9253ab 100644 --- a/docs/MUSICGEN_STYLE.md +++ b/docs/MUSICGEN_STYLE.md @@ -131,7 +131,7 @@ model.set_style_conditioner_params( melody, sr = torchaudio.load('./assets/electronic.mp3') descriptions = ["8-bit old video game music", "Chill lofi remix", "80s New wave with synthesizer"] -wav = model.generate_with_chroma(descriptions=["8-bit old video game music"], +wav = model.generate_with_chroma(descriptions=descriptions, melody[None].expand(3, -1, -1), sr) # generates 3 samples. for idx, one_wav in enumerate(wav):