diff --git a/.ci/docker/ci_commit_pins/optimum-executorch.txt b/.ci/docker/ci_commit_pins/optimum-executorch.txt index c58d8f9977b..d80136a7df3 100644 --- a/.ci/docker/ci_commit_pins/optimum-executorch.txt +++ b/.ci/docker/ci_commit_pins/optimum-executorch.txt @@ -1 +1 @@ -f8aa919593cc51301ade73a2ee5491582521ab80 +5bf1aeb587e9b1f3572b0bd60265c5dafd007b73 diff --git a/.ci/scripts/export_model_artifact.sh b/.ci/scripts/export_model_artifact.sh index b4a36383b1c..4057e491c9a 100755 --- a/.ci/scripts/export_model_artifact.sh +++ b/.ci/scripts/export_model_artifact.sh @@ -181,6 +181,8 @@ fi DEVICE_ARG="" if [ "$DEVICE" = "cuda" ] || [ "$DEVICE" = "cuda-windows" ]; then DEVICE_ARG="--device cuda" +elif [ "$DEVICE" = "metal" ]; then + DEVICE_ARG="--device mps" fi optimum-cli export executorch \