3 Commits

Author	SHA1	Message	Date
Aryan	cd991d1e1a	Fix TorchAO related bugs; revert device_map changes (#10371) ... * Revert "Add support for sharded models when TorchAO quantization is enabled (#10256)" This reverts commit 41ba8c0bf6. * update tests * udpate * update * update * update device map tests * apply review suggestions * update * make style * fix * update docs * update tests * update workflow * update * improve tests * allclose tolerance * Update src/diffusers/models/modeling_utils.py Co-authored-by: Sayak Paul <spsayakpaul@gmail.com> * Update tests/quantization/torchao/test_torchao.py Co-authored-by: Sayak Paul <spsayakpaul@gmail.com> * improve tests * fix * update correct slices --------- Co-authored-by: Sayak Paul <spsayakpaul@gmail.com>	2024-12-25 15:37:49 +05:30
Sayak Paul	6a970a45c5	[docs] fix: torchao example. (#10278) ... fix: torchao example.	2024-12-23 11:03:50 +05:30
Aryan	9f00c617a0	[core] TorchAO Quantizer (#10009) ... * torchao quantizer --------- Co-authored-by: Sayak Paul <spsayakpaul@gmail.com> Co-authored-by: Steven Liu <59462357+stevhliu@users.noreply.github.com>	2024-12-16 13:35:40 -10:00