fix

Author: a-r-r-o-w

src/diffusers/quantizers/torchao/torchao_quantizer.py

@@ -35,28 +35,28 @@
     import torch
     import torch.nn as nn
 
+    SUPPORTED_TORCH_DTYPES_FOR_QUANTIZATION = (
+        # At the moment, only int8 is supported for integer quantization dtypes.
+        # In Torch 2.6, int1-int7 will be introduced, so this can be visited in the future
+        # to support more quantization methods, such as intx_weight_only.
+        torch.int8,
+        torch.float8_e4m3fn,
+        torch.float8_e5m2,
+        torch.uint1,
+        torch.uint2,
+        torch.uint3,
+        torch.uint4,
+        torch.uint5,
+        torch.uint6,
+        torch.uint7,
+    )
+
 if is_torchao_available():
     from torchao.quantization import quantize_
 
 
 logger = logging.get_logger(__name__)
 
-SUPPORTED_TORCH_DTYPES_FOR_QUANTIZATION = (
-    # At the moment, only int8 is supported for integer quantization dtypes.
-    # In Torch 2.6, int1-int7 will be introduced, so this can be visited in the future
-    # to support more quantization methods, such as intx_weight_only.
-    torch.int8,
-    torch.float8_e4m3fn,
-    torch.float8_e5m2,
-    torch.uint1,
-    torch.uint2,
-    torch.uint3,
-    torch.uint4,
-    torch.uint5,
-    torch.uint6,
-    torch.uint7,
-)
-
 
 def _quantization_type(weight):
     from torchao.dtypes import AffineQuantizedTensor