Kernel Crash on `get_pretrained`

#5
by ngupta-microsoft - opened

I am trying to run MusicGen locally with facebook/musicgen-melody. I keep running into an error where the kernel is unexpectedly killed when running MusicGen.get_pretrained('facebook/musicgen-melody').

OS info (running in container):
Distributor ID: Debian
Description: Debian GNU/Linux 12 (bookworm)
Release: 12
Codename: bookworm

Host machine:
Machine: Macbook Pro
Chip: Apple M1 Pro
macOS: Ventura 13.5 (22G74)

I have audiocraft and ffmpeg installed.

Python 3.10.13 (main, Sep  7 2023, 13:10:41) [GCC 12.2.0] on linux
Type "help", "copyright", "credits" or "license" for more information.
>>> from audiocraft.models import MusicGen
>>> model = MusicGen.get_pretrained('facebook/musicgen-melody')
Downloading state_dict.bin: 100%|β–ˆβ–ˆβ–ˆ...β–ˆβ–ˆβ–ˆ| 2.77G/2.77G [03:08<00:00, 14.7MB/s]
Killed

Any ideas as to what may cause this?

Sign up or log in to comment