Aryan
|
12184f4015
|
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 11:31:21 +01:00 |
|
Aryan
|
02c777c065
|
[tests] Refactor TorchAO serialization fast tests (#10271)
refactor
|
2024-12-23 11:04:57 +05:30 |
|
Aryan
|
ffc0eaab6d
|
Bump minimum TorchAO version to 0.7.0 (#10293)
* bump min torchao version to 0.7.0
* update
|
2024-12-23 11:03:04 +05:30 |
|
Aryan
|
41ba8c0bf6
|
Add support for sharded models when TorchAO quantization is enabled (#10256)
* add sharded + device_map check
|
2024-12-19 15:42:20 -10:00 |
|
Aryan
|
1524781b88
|
[tests] Remove/rename unsupported quantization torchao type (#10263)
update
|
2024-12-17 21:43:15 +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 |
|