Skip to content

Commit 7451b84

Browse files
cccclaifacebook-github-bot
authored andcommitted
Enable conv_former model in CI (#12930)
Summary: Add conv_former to CI Rollback Plan: Differential Revision: D79120806
1 parent ae99c04 commit 7451b84

File tree

2 files changed

+5
-2
lines changed

2 files changed

+5
-2
lines changed

.ci/scripts/test_model.sh

Lines changed: 4 additions & 1 deletion
Original file line numberDiff line numberDiff line change
@@ -196,6 +196,9 @@ test_model_with_qnn() {
196196
EXPORT_SCRIPT=bert
197197
elif [[ "${MODEL_NAME}" == "distilbert" ]]; then
198198
EXPORT_SCRIPT=distilbert
199+
elif [[ "${MODEL_NAME}" == "conv_former" ]]; then
200+
EXPORT_SCRIPT=conv_former
201+
EXTRA_FLAGS="--dataset imagenet-mini/val"
199202
elif [[ "${MODEL_NAME}" == "eurobert" ]]; then
200203
EXPORT_SCRIPT=eurobert
201204
else
@@ -212,7 +215,7 @@ test_model_with_qnn() {
212215
"dl3"|"mv3"|"mv2"|"ic4"|"ic3"|"vit"|"mb"|"w2l")
213216
SCRIPT_FOLDER=scripts
214217
;;
215-
"albert"|"bert"|"distilbert")
218+
"albert"|"bert"|"distilbert" | "conv_former")
216219
pip install evaluate
217220
SCRIPT_FOLDER=oss_scripts
218221
# Bert models running in 16bit will encounter op validation fail on some operations,

.github/workflows/trunk.yml

Lines changed: 1 addition & 1 deletion
Original file line numberDiff line numberDiff line change
@@ -525,7 +525,7 @@ jobs:
525525
strategy:
526526
matrix:
527527
dtype: [fp32]
528-
model: [dl3, mv3, mv2, ic4, ic3, vit, mb, w2l]
528+
model: [dl3, mv3, mv2, ic4, ic3, vit, mb, w2l, conv_former]
529529
fail-fast: false
530530
with:
531531
runner: linux.2xlarge

0 commit comments

Comments
 (0)