Apple Silicon / MPS support?

lucidrains / BS-RoFormer

Implementation of Band Split Roformer, SOTA Attention network for music source separation out of ByteDance AI Labs

MIT License

433 stars 16 forks source link

Open dj-nuo opened 4 months ago

dj-nuo commented 4 months ago

Hi! Amazing work! Any chance it can run on MPS (Metal Performance Shaders) - Apple GPU?