Update

[ghstack-poisoned]
pytorch · Jan 22, 2025 · 0bc1899 · 0bc1899
1 parent 76ae4bc
commit 0bc1899
Showing 1 changed file with 3 additions and 3 deletions.
diff --git a/torchao/dtypes/affine_quantized_tensor.py b/torchao/dtypes/affine_quantized_tensor.py
@@ -6,18 +6,18 @@
 
 from torchao.dtypes.utils import AQTTensorImpl, Layout, PlainLayout
 from torchao.quantization.quant_primitives import (
+    FP8_TYPES,
+    MappingType,
+    ZeroPointDomain,
     choose_qparams_affine,
     choose_qparams_affine_float8,
     choose_qparams_affine_floatx,
     choose_qparams_and_quantize_affine_hqq,
     dequantize_affine,
     dequantize_affine_floatx,
-    FP8_TYPES,
-    MappingType,
     quantize_affine,
     quantize_affine_float8,
     quantize_affine_floatx,
-    ZeroPointDomain,
 )
 from torchao.utils import TORCH_VERSION_AT_LEAST_2_5, TorchAOBaseTensor