Update

[ghstack-poisoned]
pytorch · Feb 11, 2025 · bb253ef · bb253ef
2 parents 1c9c39f + 1ff1f6e
commit bb253ef
Showing 1 changed file with 1 addition and 1 deletion.
diff --git a/torchao/quantization/quant_api.py b/torchao/quantization/quant_api.py
@@ -884,11 +884,11 @@ def _int4_weight_only_transform(
     return module
 
 
+@dataclass
 class Int8WeightOnlyConfig(AOBaseConfig):
     """
     Configuration for applying int8 weight-only symmetric per-channel quantization to linear layers.
     """
-
     group_size: Optional[int] = None