Add support for sharded models when TorchAO quantization is enabled#10256

yiyixuxu merged 10 commits intomainhuggingface/diffusers:mainfrom

torchao-error-on-sharded-device-maphuggingface/diffusers:torchao-error-on-sharded-device-mapCopy head branch name to clipboard