add feature for loading multiple checkpoints