How to avert 'loading checkpoint shards'?

You can load the model into a local server, Flask for example.