from audiocraft.models import MusicGen, AudioGen

MusicGen.get_pretrained("facebook/musicgen-melody")