Views
No views yet
pip install --upgrade pip
pip install --upgrade transformers scipy1from transformers import AutoProcessor, MusicgenForConditionalGeneration
2import scipy
3
4processor = AutoProcessor.from_pretrained("facebook/musicgen-small")
5model = MusicgenForConditionalGeneration.from_pretrained("bfh-genai/meditation-musicgen")
6
7relaxing_description = 'Peaceful meditation background sound'
8
9conditioned_text_input = processor(text=relaxing_description, padding=True, return_tensors="pt")
10
11audio_value = model.generate(**conditioned_text_input,
12 do_sample=True,
13 guidance_scale=3, # Value >1, best results achieved with 3.
14 max_new_tokens=256 # 256 ^= 5 seconds of audio.
15 )
16
17scipy.io.wavfile.write(f"my_audio.wav", rate=32_000, data=audio_values[0, 0].numpy())