mirror of
https://github.com/huggingface/transformers.git
synced 2025-07-31 02:02:21 +06:00
[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:
parent
1d3f35f30a
commit
068b663f90
@ -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()
|
||||
|
Loading…
Reference in New Issue
Block a user