transformers
ae9dd02e
- Fix incorrect accelerator device handling for MPS in `TrainingArguments` (#31812)
Go
Login via GitHub
Home
Pricing
FAQ
Install
Login
via GitHub
Commit
View On
GitHub
Commit
1 year ago
Fix incorrect accelerator device handling for MPS in `TrainingArguments` (#31812) * Fix wrong acclerator device setup when using MPS * More robust TrainingArguments MPS handling * Update training_args.py * Cleanup
References
#31812 - Fix incorrect accelerator device handling for MPS in `TrainingArguments`
Author
andstor
Parents
4879ac2b
Loading