a few fix for shard checkpoints (#8656)

Commit

1 year ago

a few fix for shard checkpoints (#8656) fix Co-authored-by: yiyixuxu <yixu310@gmail,com>