allow to load from diloco ckpt from a different diloco shard #91
Description
when starting from a diloco ckpt you might want to use a different diloco rank
{{ message }}
when starting from a diloco ckpt you might want to use a different diloco rank