Cannot assign BERT model in class field

Question

I have a class that extends the LightningModule class of Pytorch Lightning and has a field bert_model. I am trying to perform the following