andrewor14 fdcfdbfd22 Fix TorchAoConfig not JSON serializable (#36206) ... **Summary:** TorchAoConfig optionally contains a `torchao.dtypes.Layout` object which is a dataclass and not JSON serializable, and so the following fails: ``` import json from torchao.dtypes import TensorCoreTiledLayout from transformers import TorchAoConfig config = TorchAoConfig("int4_weight_only", layout=TensorCoreTiledLayout()) config.to_json_string() json.dumps(config.to_dict()) ``` This also causes `quantized_model.save_pretrained(...)` to fail because the first step of this call is to JSON serialize the config. Fixes https://github.com/pytorch/ao/issues/1704. **Test Plan:** python tests/quantization/torchao_integration/test_torchao.py -k test_json_serializable Co-authored-by: Mohamed Mekkouri <93391238+MekkCyber@users.noreply.github.com> Co-authored-by: Marc Sun <57196510+SunMarc@users.noreply.github.com>		2025-02-18 11:05:42 +01:00
..
__init__.py	Add TorchAOHfQuantizer (#32306)	2024-08-14 16:14:24 +02:00
test_torchao.py	Fix TorchAoConfig not JSON serializable (#36206)	2025-02-18 11:05:42 +01:00