From bad78605b194d8ebc941d4dc5a3d17a92f096586 Mon Sep 17 00:00:00 2001 From: Chen Lai Date: Mon, 4 Aug 2025 11:00:34 -0700 Subject: [PATCH 1/2] Enable conv_former model in CI Summary: Add conv_former to CI Differential Revision: D79120806 --- .ci/scripts/test_model.sh | 5 ++++- .github/workflows/trunk.yml | 2 +- 2 files changed, 5 insertions(+), 2 deletions(-) diff --git a/.ci/scripts/test_model.sh b/.ci/scripts/test_model.sh index 5d9f694b0b6..1eed48f4535 100755 --- a/.ci/scripts/test_model.sh +++ b/.ci/scripts/test_model.sh @@ -199,6 +199,9 @@ test_model_with_qnn() { EXPORT_SCRIPT=albert elif [[ "${MODEL_NAME}" == "bert" ]]; then EXPORT_SCRIPT=bert + elif [[ "${MODEL_NAME}" == "conv_former" ]]; then + EXPORT_SCRIPT=conv_former + EXTRA_FLAGS="--dataset imagenet-mini/val" elif [[ "${MODEL_NAME}" == "cvt" ]]; then EXPORT_SCRIPT=cvt elif [[ "${MODEL_NAME}" == "distilbert" ]]; then @@ -238,7 +241,7 @@ test_model_with_qnn() { "cvt"|"dit"|"focalnet"|"mobilevit_v2"|"pvt"|"swin") SCRIPT_FOLDER=oss_scripts ;; - "albert"|"bert"|"distilbert"|"roberta"|"efficientnet"|"mobilevit_v1") + "albert"|"bert"|"conv_former"|"distilbert"|"roberta"|"efficientnet"|"mobilevit_v1") pip install evaluate SCRIPT_FOLDER=oss_scripts # 16bit models will encounter op validation fail on some operations, diff --git a/.github/workflows/trunk.yml b/.github/workflows/trunk.yml index 0e22f3defe1..19a0963598f 100644 --- a/.github/workflows/trunk.yml +++ b/.github/workflows/trunk.yml @@ -568,7 +568,7 @@ jobs: strategy: matrix: dtype: [fp32] - model: [dl3, mv3, mv2, ic4, ic3, vit, mb, w2l] + model: [dl3, mv3, mv2, ic4, ic3, vit, mb, w2l, conv_former] fail-fast: false with: runner: linux.2xlarge From 4c69f90529f6c6c9a61c8a40b72a2f0dbdb86f56 Mon Sep 17 00:00:00 2001 From: Chen Lai Date: Mon, 4 Aug 2025 12:41:41 -0700 Subject: [PATCH 2/2] Enable dinov2 model in CI Differential Revision: D79574148 --- .ci/scripts/test_model.sh | 5 ++++- 1 file changed, 4 insertions(+), 1 deletion(-) diff --git a/.ci/scripts/test_model.sh b/.ci/scripts/test_model.sh index 1eed48f4535..b22d7db56ac 100755 --- a/.ci/scripts/test_model.sh +++ b/.ci/scripts/test_model.sh @@ -204,6 +204,9 @@ test_model_with_qnn() { EXTRA_FLAGS="--dataset imagenet-mini/val" elif [[ "${MODEL_NAME}" == "cvt" ]]; then EXPORT_SCRIPT=cvt + elif [[ "${MODEL_NAME}" == "dino_v2" ]]; then + EXPORT_SCRIPT=dino_v2 + EXTRA_FLAGS="--dataset imagenet-mini/val" elif [[ "${MODEL_NAME}" == "distilbert" ]]; then EXPORT_SCRIPT=distilbert elif [[ "${MODEL_NAME}" == "dit" ]]; then @@ -241,7 +244,7 @@ test_model_with_qnn() { "cvt"|"dit"|"focalnet"|"mobilevit_v2"|"pvt"|"swin") SCRIPT_FOLDER=oss_scripts ;; - "albert"|"bert"|"conv_former"|"distilbert"|"roberta"|"efficientnet"|"mobilevit_v1") + "albert"|"bert"|"conv_former"|"dino_v2"|"distilbert"|"roberta"|"efficientnet"|"mobilevit_v1") pip install evaluate SCRIPT_FOLDER=oss_scripts # 16bit models will encounter op validation fail on some operations,