Fix GPTQ import error after torchao refactor

Summary: Fix broken import after pytorch/ao#275 Reviewed By: jerryzh168 Differential Revision: D57888168
jerryzh168 · May 29, 2024 · 8058ade · 8058ade
1 parent c665c17
commit 8058ade
Showing 1 changed file with 1 addition and 1 deletion.
diff --git a/examples/models/llama2/source_transformation/quantize.py b/examples/models/llama2/source_transformation/quantize.py
@@ -96,7 +96,7 @@ def quantize(
         if calibration_tasks is None:
             calibration_tasks = ["wikitext"]
 
-        from torchao.quantization.GPTQ import InputRecorder
+        from torchao._eval import InputRecorder
         from torchao.quantization.quant_api import Int8DynActInt4WeightGPTQQuantizer
 
         if tokenizer_path is None: