diff --git a/.ci/scripts/test_model.sh b/.ci/scripts/test_model.sh index 5d9f694b0b6..b22d7db56ac 100755 --- a/.ci/scripts/test_model.sh +++ b/.ci/scripts/test_model.sh @@ -199,8 +199,14 @@ 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}" == "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 @@ -238,7 +244,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"|"dino_v2"|"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