Spaces:

shangrilar
/

soojinchoi_test

Paused

shangrilar commited on Mar 19, 2024

Commit

a8dea68

verified ·

1 Parent(s): 1f85a54

Update utils.py

Files changed (1) hide show

utils.py CHANGED Viewed

@@ -27,7 +27,7 @@ from audiocraft.data.audio import audio_write
 from audiocraft.models import MusicGen
 # Using small model, better results would be obtained with `medium` or `large`.
-model = MusicGen.get_pretrained('melody')
 model.set_generation_params(
     use_sampling=True,
     top_k=250,

 from audiocraft.models import MusicGen
 # Using small model, better results would be obtained with `medium` or `large`.
+model = MusicGen.get_pretrained('facebook/musicgen-melody')
 model.set_generation_params(
     use_sampling=True,
     top_k=250,