Loading checkpoint shards: 0%| | 0/3 [00:00 outputs/uniform_control_materialize/results.json