diff --git a/torchao/quantization/quant_api.py b/torchao/quantization/quant_api.py index 139b14cf3f..c7a87e72b8 100644 --- a/torchao/quantization/quant_api.py +++ b/torchao/quantization/quant_api.py @@ -2317,7 +2317,7 @@ def _fpx_weight_only_transform( @dataclass class ModuleFqnToConfig(AOBaseConfig): - """Per module configurations for torchao quantize_ API + r"""Per module configurations for torchao quantize_ API Args: `module_fqn_to_config`: typing.OrderedDict[str, Optional[AOBaseConfig]]: an