Views
No views yet
facebook/musicgen-medium, specialized in generating high-quality Jazz, Soul, and R&B music with strong structural consistency and long-form stability.1import torch
2from transformers import MusicgenForConditionalGeneration, AutoProcessor
3from peft import PeftModel
4
5model_id = "facebook/musicgen-medium"
6lora_id = "ryanbw/musicgen-medium-jazz-soul-lora"
7
8model = MusicgenForConditionalGeneration.from_pretrained(model_id, torch_dtype=torch.float16).to("cuda")
9model = PeftModel.from_pretrained(model, lora_id)
10processor = AutoProcessor.from_pretrained(model_id)
11
12# Example Prompt
13prompt = "A professional recording of Bossa Nova rhythm, Latin Jazz influence, 129 BPM, Key of D, smooth groove, studio quality, structured musical loop."
14inputs = processor(text=[prompt], return_tensors="pt").to("cuda")
15
16# Generate ~20 seconds of audio
17audio = model.generate(**inputs, max_new_tokens=1000, guidance_scale=3.5, temperature=0.85)