The error message claims only 130.87 MiB are free.
What is nvidia-smi showing?
Besides the parameters of the model, a large portion of the memory usage might be due to the forward activations, gradients etc.