Enable using HF PARQ checkpoints in torchao (#2985)

metascroy · web-flow · commit 93030e750186 · 2025-09-11T17:10:58.000-07:00
up
diff --git a/torchao/core/config.py b/torchao/core/config.py
@@ -193,6 +193,7 @@ def config_to_dict(config: AOBaseConfig) -> Dict[str, Any]:
     "torchao.sparsity.sparse_api",
     "torchao.prototype.quantization",
     "torchao.prototype.mx_formats",
+    "torchao.prototype.parq",
     "torchao.dtypes",
     "torchao.prototype.awq",
     "torchao.quantization.quantize_.common",
diff --git a/torchao/prototype/parq/__init__.py b/torchao/prototype/parq/__init__.py
@@ -20,3 +20,8 @@
     UnifQuantizer,
     UnifTorchaoQuantizer,
 )
+from .quant.config_torchao import StretchedIntxWeightConfig
+
+__all__ = [
+    "StretchedIntxWeightConfig",
+]

Original file line number	Diff line number	Diff line change
`@@ -20,3 +20,8 @@`
`20`	`20`	`UnifQuantizer,`
`21`	`21`	`UnifTorchaoQuantizer,`
`22`	`22`	`)`
	`23`	`+from .quant.config_torchao import StretchedIntxWeightConfig`
	`24`	`+`
	`25`	`+__all__ = [`
	`26`	`+ "StretchedIntxWeightConfig",`
	`27`	`+]`