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

See if we should use map_location=torch.device("cpu") when loading weights in various places #84

Open

favyen2 opened this issue Oct 16, 2024 · 0 comments · May be fixed by #87

Assignees

Collaborator

favyen2 commented Oct 16, 2024

Probably makes sense to. This causes error when trying to run model with only CPU.

favyen2 self-assigned this

favyen2 linked a pull request

that will close this issue

Always load weights to CPU. #87

Open

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