From 7d546824f5dabbd1e4f62d277a00b9b6738b3207 Mon Sep 17 00:00:00 2001 From: QQSong Date: Mon, 24 Jun 2024 14:48:28 -0700 Subject: [PATCH] Update benchmark_aq.py --- benchmarks/benchmark_aq.py | 2 +- 1 file changed, 1 insertion(+), 1 deletion(-) diff --git a/benchmarks/benchmark_aq.py b/benchmarks/benchmark_aq.py index 09fe1bc45..fc19d1ab8 100644 --- a/benchmarks/benchmark_aq.py +++ b/benchmarks/benchmark_aq.py @@ -5,7 +5,7 @@ Int8WeightOnlyQuantizedLinearWeight, Int4WeightOnlyQuantizedLinearWeight, ) -from torchao.quantization.utils import ( +from torchao.utils import ( TORCH_VERSION_AFTER_2_4, ) from torchao.quantization.quant_api import (