My code is as follows:
import tensorflow as tf from transformers import RobertaConfig, TFRobertaMainLayer # 1. Create a class to be able to use fit() class