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.

Sign up for GitHub

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

Jump to bottom

what do we need to set the device parameter when we use macos ? #53

Open

hvacking opened this issue Apr 21, 2024 · 1 comment

hvacking commented Apr 21, 2024

It seems that it is not correct when we set from 'device = "cuda' to device = "cpu"?

Contributor

ashok-arjun commented Apr 21, 2024

I think it should be torch.device("mps")

Sign up for free to join this conversation on GitHub. Already have an account? Sign in to comment