Fix GPTQ import error after torchao refactor (pytorch#3760)

Summary:

Fix broken import after pytorch/ao#275

Reviewed By: jerryzh168

Differential Revision: D57888168

examples/models/llama2/source_transformation/quantize.py

@@ -96,7 +96,13 @@ def quantize(
         if calibration_tasks is None:
             calibration_tasks = ["wikitext"]
 
-        from torchao.quantization.GPTQ import InputRecorder
+        try:
+            # torchao 0.3+
+            # pyre-ignore
+            from torchao._eval import InputRecorder
+        except ImportError:
+            from torchao.quantization.GPTQ import InputRecorder
+
         from torchao.quantization.quant_api import Int8DynActInt4WeightGPTQQuantizer
 
         if tokenizer_path is None: