Skip to content

Commit 068b663

Browse files
SunMarcMekkCyber
andauthored
[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>
1 parent 1d3f35f commit 068b663

File tree

1 file changed

+1
-1
lines changed

1 file changed

+1
-1
lines changed

src/transformers/utils/quantization_config.py

Lines changed: 1 addition & 1 deletion
Original file line numberDiff line numberDiff line change
@@ -1595,7 +1595,7 @@ def _get_torchao_quant_type_to_method(self):
15951595
"autoquant": autoquant,
15961596
}
15971597

1598-
def get_quantize_config(self):
1598+
def get_apply_tensor_subclass(self):
15991599
"""Create the appropriate quantization method based on configuration."""
16001600
if isinstance(self.quant_type, str):
16011601
methods = self._get_torchao_quant_type_to_method()

0 commit comments

Comments
 (0)