transformers/tests/quantization/torchao_integration
Benjamin Bossan 5e1fd4e204
FIX: Broken repr of TorchAoConfig (#34560)
FIX Broken repr of TorchAoConfig

The __repr__ method references a non-existent self.kwargs. This is now
fixed.

There does not appear to be a uniform way of defining __repr__ for
quantization configs. I copied the method as implemented for HQQ:

e2ac16b28a/src/transformers/utils/quantization_config.py (L285-L287)
2024-11-05 10:26:13 +01:00
..
__init__.py Add TorchAOHfQuantizer (#32306) 2024-08-14 16:14:24 +02:00
test_torchao.py FIX: Broken repr of TorchAoConfig (#34560) 2024-11-05 10:26:13 +01:00