search

acids-ircam / RAVE

Official implementation of the RAVE model: a Realtime Audio Variational autoEncoder
Other
1.35k stars 184 forks source link

Rave on MacOSX M2 #274

Closed Federico8691 closed 11 months ago

Federico8691 commented 11 months ago

Hey Axel,

does Rave works on apple silicon now as well?

Thanks!

domkirke commented 11 months ago

Yes

devstermarts commented 11 months ago

Hi Axel, does training on Apple GPUs work out of the box? I'm still running into 'Training on mac is not available yet. Use --gpu -1 to train on CPU (not recommended).' - v. 2.3.1. is installed. Torch version is 2.1.2

Thanks! Martin

domkirke commented 11 months ago

This is not a problem of RAVE, this is a problem with current M2 devices that are not powerful enough to correctly train a model. But you can try :)

devstermarts commented 11 months ago

Not sure if I understand correctly; training on CPU is obviously senseless, but shouldn't Torch's support for MPS make training on Apple MX feasible in general? Tampering with train.py at least gets me to MPS confirmed as GPU being used, but I'm running into errors with function calls in Lightning. Anyway, will keep an eye out on that topic.