MMBTForClassification to torchscript

Has anyone transformed MMBTForClassification model to torchscript before? I got error when I include return_dict = False. Would love to have some guidance on this

Best,
Henry