BERT model loading not working with pytorch 1.3.1-eia container

Question

I have a custom python file for inference in which I have implemented the functions model_fn, input_fn, predict_fn and output_fn