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?