Pytorch error when training DeepSeek R1

AutoTrain is currently experiencing a lot of errors… @abhishek