File tree Expand file tree Collapse file tree 1 file changed +4
-1
lines changed Expand file tree Collapse file tree 1 file changed +4
-1
lines changed Original file line number Diff line number Diff 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,
You can’t perform that action at this time.
0 commit comments