extend TorchAoTest::test_model_memory_usage to other platform#12768

sayakpaul merged 3 commits intohuggingface:mainhuggingface/diffusers:mainfrom

sywangyi:torchao_fix_memsywangyi/diffusers:torchao_fix_memCopy head branch name to clipboard