transformers
2aa2a144
- Make tensor device correct when ACCELERATE_TORCH_DEVICE is defined (#31751)
Go
Login via GitHub
Home
Pricing
FAQ
Install
Login
via GitHub
Commit
View On
GitHub
Commit
1 year ago
Make tensor device correct when ACCELERATE_TORCH_DEVICE is defined (#31751) return correct device when ACCELERATE_TORCH_DEVICE is defined
References
#31751 - Make tensor device correct when ACCELERATE_TORCH_DEVICE is defined
Author
kiszk
Parents
8c5c180d
Loading