Ahmadzei's picture
added 3 more tables for large emb model
5fa1a76
pip install torch torchvision torchaudio
[TrainingArguments] uses the mps device by default if it's available which means you don't need to explicitly set the device.