Update model loading logic and add support for multiple checkpoints 7576408 Dan Bochman commited on Aug 26