transformers
Fix incorrect accelerator device handling for MPS in `TrainingArguments`
#31812

Merged

Fix incorrect accelerator device handling for MPS in `TrainingArguments` #31812

amyeroberts merged 4 commits into huggingface:main from andstor:patch-7

Fix wrong acclerator device setup when using MPS

719215c8

More robust TrainingArguments MPS handling

15604306

Update training_args.py

20f19072

Cleanup

86ff86ec

amyeroberts approved these changes on 2024-07-08

SunMarc approved these changes on 2024-07-08

amyeroberts merged ae9dd02e into main 1 year ago

Reviewers

SunMarc

amyeroberts

Assignees

No one assigned

Labels

None yet

Milestone

No milestone