-
-
Notifications
You must be signed in to change notification settings - Fork 2.1k
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
[Bug]: error when training using mps #3506
Comments
I got past that by using a newer version of torch (2.5.0) and transformers (4.43.3). Using Flair version 0.13.1 or 0.14.0 gives me the following issue when training a model ... Traceback (most recent call last): I validate my instance of Torch is good and accepts "mps" using sample code from here - https://github.com/mrdbourke/pytorch-apple-silicon Is MPS (Metal Performance Shader) built? True |
Describe the bug
When setting
flair.device
tomps
, the following error is thrown during training:To Reproduce
Expected behavior
Torch's mps support should be usable via flair.
Logs and Stack traces
No response
Screenshots
No response
Additional Context
No response
Environment
Versions:
Flair
0.13.1
Pytorch
2.3.1
Transformers
4.42.4
GPU
False
The text was updated successfully, but these errors were encountered: