Skip to content

Commit 4c69f90

Browse files
cccclaifacebook-github-bot
authored andcommitted
Enable dinov2 model in CI
Differential Revision: D79574148
1 parent bad7860 commit 4c69f90

File tree

1 file changed

+4
-1
lines changed

1 file changed

+4
-1
lines changed

.ci/scripts/test_model.sh

Lines changed: 4 additions & 1 deletion
Original file line numberDiff line numberDiff line change
@@ -204,6 +204,9 @@ test_model_with_qnn() {
204204
EXTRA_FLAGS="--dataset imagenet-mini/val"
205205
elif [[ "${MODEL_NAME}" == "cvt" ]]; then
206206
EXPORT_SCRIPT=cvt
207+
elif [[ "${MODEL_NAME}" == "dino_v2" ]]; then
208+
EXPORT_SCRIPT=dino_v2
209+
EXTRA_FLAGS="--dataset imagenet-mini/val"
207210
elif [[ "${MODEL_NAME}" == "distilbert" ]]; then
208211
EXPORT_SCRIPT=distilbert
209212
elif [[ "${MODEL_NAME}" == "dit" ]]; then
@@ -241,7 +244,7 @@ test_model_with_qnn() {
241244
"cvt"|"dit"|"focalnet"|"mobilevit_v2"|"pvt"|"swin")
242245
SCRIPT_FOLDER=oss_scripts
243246
;;
244-
"albert"|"bert"|"conv_former"|"distilbert"|"roberta"|"efficientnet"|"mobilevit_v1")
247+
"albert"|"bert"|"conv_former"|"dino_v2"|"distilbert"|"roberta"|"efficientnet"|"mobilevit_v1")
245248
pip install evaluate
246249
SCRIPT_FOLDER=oss_scripts
247250
# 16bit models will encounter op validation fail on some operations,

0 commit comments

Comments
 (0)