diff --git a/src/diffusers/quantizers/quantization_config.py b/src/diffusers/quantizers/quantization_config.py index 440ef2bf6230..4fac8dd3829f 100644 --- a/src/diffusers/quantizers/quantization_config.py +++ b/src/diffusers/quantizers/quantization_config.py @@ -47,7 +47,7 @@ class QuantizationMethod(str, Enum): TORCHAO = "torchao" -if is_torchao_available: +if is_torchao_available(): from torchao.quantization.quant_primitives import MappingType class TorchAoJSONEncoder(json.JSONEncoder):