Loading checkpoint shards:   0%|          | 0/5 [00:00<?, ?it/s]
Loading checkpoint shards:  20%|██        | 1/5 [00:03<00:13,  3.37s/it]
Loading checkpoint shards:  40%|████      | 2/5 [00:05<00:08,  2.81s/it]
Loading checkpoint shards:  60%|██████    | 3/5 [00:08<00:05,  2.89s/it]
Loading checkpoint shards:  80%|████████  | 4/5 [00:11<00:02,  2.80s/it]
Loading checkpoint shards: 100%|██████████| 5/5 [00:11<00:00,  1.93s/it]
Loading checkpoint shards: 100%|██████████| 5/5 [00:11<00:00,  2.36s/it]