Exporting a Bert-based PyTorch model to CoreML. How can I make the CoreML model work for any input? I use the code below to export a Bert-based PyTorch model to CoreML.