megablocks
megablocks copied to clipboard
Unsharding scripts for megablocks models
The base Megatron-LM repo provides unsharding scripts for the models which can be used after training. I didn't find any such scripts in the repo. Would it be possible to provide the same?