Add MPS support #7

Open
opened 2025-10-14 15:38:11 -06:00 by navan · 0 comments
Owner

Originally created by @okalex on 2/22/2025

This PR adds support for apple silicon. torch.scatter wasn't supported with imaginary numbers on MPS, so I changed that according to a chatgpt suggestion (sorry, I'm not much of a python/pytorch developer), and can confirm that it works on my machine. Let me know if there's anything else you'd like to see here.

*Originally created by @okalex on 2/22/2025* This PR adds support for apple silicon. `torch.scatter` wasn't supported with imaginary numbers on MPS, so I changed that according to a chatgpt suggestion (sorry, I'm not much of a python/pytorch developer), and can confirm that it works on my machine. Let me know if there's anything else you'd like to see here.
Sign in to join this conversation.
No labels
No milestone
No project
No assignees
1 participant
Notifications
Due date
The due date is invalid or out of range. Please use the format "yyyy-mm-dd".

No due date set.

Dependencies

No dependencies set.

Reference: github/Mel-Band-Roformer-Vocal-Model#7
No description provided.