add support for loading multiple checkpoints