Skip to content
Open
Show file tree
Hide file tree
Changes from all commits
Commits
File filter

Filter by extension

Filter by extension


Conversations
Failed to load comments.
Loading
Jump to
Jump to file
Failed to load files.
Loading
Diff view
Diff view
8 changes: 7 additions & 1 deletion .ci/scripts/test_model.sh
Original file line number Diff line number Diff line change
Expand Up @@ -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
Expand Down Expand Up @@ -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,
Expand Down
2 changes: 1 addition & 1 deletion .github/workflows/trunk.yml
Original file line number Diff line number Diff line change
Expand Up @@ -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
Expand Down
Loading