training_args = TrainingArguments( no_cuda=True, # )
adding this works for me, this indicates Use CPU since MPS may have compatibility issues.