Skip to content

Commit 66929fd

Browse files
committed
Update library path
1 parent d8bbe70 commit 66929fd

File tree

1 file changed

+4
-4
lines changed
  • examples/qualcomm/oss_scripts/llama

1 file changed

+4
-4
lines changed

examples/qualcomm/oss_scripts/llama/llama.py

Lines changed: 4 additions & 4 deletions
Original file line numberDiff line numberDiff line change
@@ -19,7 +19,7 @@
1919
from multiprocessing.connection import Client
2020

2121
import torch
22-
from executorch.backends.qualcomm._passes.i64_to_i32 import I64toI32
22+
from executorch.backends.qualcomm._passes.constant_i64_to_i32 import ConstantI64toI32
2323

2424
from executorch.backends.qualcomm.partition.qnn_partitioner import QnnPartitioner
2525

@@ -553,9 +553,9 @@ def compile(args, pte_filename, tokenizer):
553553
llama_instance_list[i] = get_quant_embedding_transform(args)(
554554
llama_instance_list[i]
555555
)
556-
passes_job[I64toI32][QCOM_PASS_ARGS_KWARGS_DEFAULTS_KEY]["skip_node"] = {
557-
"tokens"
558-
}
556+
passes_job[ConstantI64toI32][QCOM_PASS_ARGS_KWARGS_DEFAULTS_KEY][
557+
"skip_node"
558+
] = {"tokens"}
559559
llama_instance_list[i] = convert_linear_to_conv2d(llama_instance_list[i])
560560
llama_instance_list[i] = SingleLlama(
561561
llama_instance_list[i].eval(), pte_filename

0 commit comments

Comments
 (0)