How to handle imbalanced classes in transformers pytorch binary classification

Question

I am working on binary text classification problem. How do i apply smote or WeightedRandomSample for the imbalance in my dataset.My code currenlty looks like this: