BERT model throws error when used in Pytorch Lightning

[sudarshan85]

OMG…thank you. That was so dumb of me.