[torchao] revert to get_apply_tensor_subclass (#36849)

* revert to old name

* empty commit

---------

Co-authored-by: Mohamed Mekkouri <93391238+MekkCyber@users.noreply.github.com>
This commit is contained in:
Marc Sun 2025-03-20 18:00:13 +01:00 committed by GitHub
parent 1d3f35f30a
commit 068b663f90
No known key found for this signature in database
GPG Key ID: B5690EEEBB952194

View File

@ -1595,7 +1595,7 @@ class TorchAoConfig(QuantizationConfigMixin):
"autoquant": autoquant,
}
def get_quantize_config(self):
def get_apply_tensor_subclass(self):
"""Create the appropriate quantization method based on configuration."""
if isinstance(self.quant_type, str):
methods = self._get_torchao_quant_type_to_method()