transformers
Fix incorrect accelerator device handling for MPS in `TrainingArguments`
#31812
Merged
Overview
Commits
4
Changes
View On GitHub
Commits
  • Fix wrong acclerator device setup when using MPS
    andstor committed 1 year ago
  • More robust TrainingArguments MPS handling
    andstor committed 1 year ago
  • Update training_args.py
    andstor committed 1 year ago
  • Cleanup
    andstor committed 1 year ago
FAQ Terms Privacy Refunds Impressum
Loading