[torchao] fix safetensors and enable loading from sharded files #41998

liangel-02 wants to merge 1 commit intohuggingface:mainhuggingface/transformers:mainfrom

liangel-02:torchao-safetensors-shardingliangel-02/transformers:torchao-safetensors-shardingCopy head branch name to clipboard