feat: save sharded safetensors#115

fabianlim merged 1 commit intofoundation-model-stack:checkpoint-utils-safetensorsfoundation-model-stack/fms-acceleration:checkpoint-utils-safetensorsfrom

willmj:save_sharded_safetensorswillmj/fms-acceleration:save_sharded_safetensorsCopy head branch name to clipboard