Skip to content

Commit 3fbefec

Browse files
committed
Added NPU in choices
1 parent 6d0cbc5 commit 3fbefec

File tree

1 file changed

+2
-2
lines changed

1 file changed

+2
-2
lines changed

examples/models/llama/export_llama_lib.py

Lines changed: 2 additions & 2 deletions
Original file line numberDiff line numberDiff line change
@@ -464,8 +464,8 @@ def build_args_parser() -> argparse.ArgumentParser:
464464
"--openvino_device",
465465
type=str,
466466
default="CPU",
467-
choices=["CPU", "GPU"],
468-
help="Specify the device for Openvino (CPU or GPU).",
467+
choices=["CPU", "GPU", "NPU"],
468+
help="Specify the device for Openvino (CPU, GPU or NPU).",
469469
)
470470

471471
parser.add_argument(

0 commit comments

Comments
 (0)