-
Notifications
You must be signed in to change notification settings - Fork 173
New issue
Have a question about this project? Sign up for a free GitHub account to open an issue and contact its maintainers and the community.
By clicking “Sign up for GitHub”, you agree to our terms of service and privacy statement. We’ll occasionally send you account related emails.
Already on GitHub? Sign in to your account
Support for M1 Mac, or non-cuda devices #11
Labels
enhancement
New feature or request
Comments
mattkanwisher
changed the title
Support for M1 Mac
Support for M1 Mac, or non-cuda devices
Mar 12, 2024
We don't run it on Mac internally, so nobody is doing this. But I guess this should be easy and just need a little bit coding. We would appreciate it if you can do this and make a pull request. |
This PR #22 |
confirmed #22 makes it run now, albeit slowly, some of the ops are still unsupported by mps by PyTorch but fall back to cpu |
Sign up for free
to join this conversation on GitHub.
Already have an account?
Sign in to comment
Seems like parts of the code explicitly call cuda functions, so you can't just switch it to MPS for Macs. Any roadmap for supporting Macs?
Io.py
error
The text was updated successfully, but these errors were encountered: