We read every piece of feedback, and take your input very seriously.
To see all available qualifiers, see our documentation.
There was an error while loading. Please reload this page.
1 parent 381ae5d commit 5a1aa1fCopy full SHA for 5a1aa1f
examples/models/llama/source_transformation/quantize.py
@@ -177,7 +177,7 @@ def quantize( # noqa C901
177
178
try:
179
# torchao 0.3+
180
- from torchao._eval import InputRecorder # pyre-fixme[21]
+ from torchao._models._eval import InputRecorder
181
except ImportError:
182
from torchao.quantization.GPTQ import InputRecorder # pyre-ignore
183
0 commit comments