Skip to content

Commit ec27667

Browse files
authored
Fix pyre issue
Differential Revision: D65078023 Pull Request resolved: #6532
1 parent 4e6976d commit ec27667

File tree

2 files changed

+4
-2
lines changed

2 files changed

+4
-2
lines changed

examples/models/llama/export_llama_lib.py

Lines changed: 3 additions & 1 deletion
Original file line numberDiff line numberDiff line change
@@ -665,14 +665,16 @@ def _export_llama(modelname, args) -> LLMEdgeManager: # noqa: C901
665665

666666
from functools import partial
667667

668+
# pyre-ignore
668669
from executorch.backends.qualcomm.quantizer.custom_annotation import (
669670
get_custom_quant_ios_dtype,
670671
)
671672

673+
# pyre-ignore
672674
tag_quant_io(
673675
builder_exported_to_edge.edge_manager.exported_program().graph_module,
674676
partial(
675-
get_custom_quant_ios_dtype,
677+
get_custom_quant_ios_dtype, # pyre-ignore
676678
builder_exported_to_edge.model.layers[
677679
0
678680
].attention.kv_cache.past_k_caches.shape,

extension/llm/export/builder.py

Lines changed: 1 addition & 1 deletion
Original file line numberDiff line numberDiff line change
@@ -219,7 +219,7 @@ def pt2e_calibrate(
219219
from executorch.examples.models.llama.eval_llama_lib import (
220220
GraphModuleEvalWrapper,
221221
)
222-
from lm_eval.evaluator import simple_evaluate # pyre-ignore[21]
222+
from lm_eval.evaluator import simple_evaluate
223223
except ImportError:
224224
raise ImportError(
225225
"Please install the llm eval dependency via examples/models/llama/install_requirements.sh"

0 commit comments

Comments
 (0)