Skip to content

Commit 77797c2

Browse files
Chia-Jung Changfacebook-github-bot
authored andcommitted
Fix bugs in executorch package (#10251)
Summary: There were bugs in `executorch`, which did not update `InputRecorder` import from `torchao` given changes in the path, which also required adding dependency to import `lm_eval`. Differential Revision: D73166222
1 parent 17cbef5 commit 77797c2

File tree

1 file changed

+1
-1
lines changed

1 file changed

+1
-1
lines changed

examples/models/llama/source_transformation/quantize.py

Lines changed: 1 addition & 1 deletion
Original file line numberDiff line numberDiff line change
@@ -164,7 +164,7 @@ def quantize( # noqa C901
164164

165165
try:
166166
# torchao 0.3+
167-
from torchao._eval import InputRecorder # pyre-fixme[21]
167+
from torchao._models._eval import InputRecorder
168168
except ImportError:
169169
from torchao.quantization.GPTQ import InputRecorder # pyre-ignore
170170

0 commit comments

Comments
 (0)