@@ -188,22 +188,6 @@ test_model_with_qnn() {
188188 EXPORT_SCRIPT=edsr
189189 # Additional deps for edsr
190190 pip install piq
191- elif [[ " ${MODEL_NAME} " == " cvt" ]]; then
192- EXPORT_SCRIPT=cvt
193- elif [[ " ${MODEL_NAME} " == " dit" ]]; then
194- EXPORT_SCRIPT=dit
195- elif [[ " ${MODEL_NAME} " == " efficientnet" ]]; then
196- EXPORT_SCRIPT=efficientnet
197- elif [[ " ${MODEL_NAME} " == " focalnet" ]]; then
198- EXPORT_SCRIPT=focalnet
199- elif [[ " ${MODEL_NAME} " == " mobilevit_v1" ]]; then
200- EXPORT_SCRIPT=mobilevit_v1
201- elif [[ " ${MODEL_NAME} " == " mobilevit_v2" ]]; then
202- EXPORT_SCRIPT=mobilevit_v2
203- elif [[ " ${MODEL_NAME} " == " pvt" ]]; then
204- EXPORT_SCRIPT=pvt
205- elif [[ " ${MODEL_NAME} " == " swin" ]]; then
206- EXPORT_SCRIPT=swin_transformer
207191 elif [[ " ${MODEL_NAME} " == " albert" ]]; then
208192 EXPORT_SCRIPT=albert
209193 elif [[ " ${MODEL_NAME} " == " bert" ]]; then
@@ -212,8 +196,6 @@ test_model_with_qnn() {
212196 EXPORT_SCRIPT=distilbert
213197 elif [[ " ${MODEL_NAME} " == " eurobert" ]]; then
214198 EXPORT_SCRIPT=eurobert
215- elif [[ " ${MODEL_NAME} " == " roberta" ]]; then
216- EXPORT_SCRIPT=roberta
217199 else
218200 echo " Unsupported model $MODEL_NAME "
219201 exit 1
@@ -228,13 +210,10 @@ test_model_with_qnn() {
228210 " dl3" |" mv3" |" mv2" |" ic4" |" ic3" |" vit" |" mb" |" w2l" )
229211 SCRIPT_FOLDER=scripts
230212 ;;
231- " cvt" |" dit" |" focalnet" |" mobilevit_v2" |" pvt" |" swin" )
232- SCRIPT_FOLDER=oss_scripts
233- ;;
234- " albert" |" bert" |" distilbert" |" roberta" |" efficientnet" |" mobilevit_v1" )
213+ " albert" |" bert" |" distilbert" )
235214 pip install evaluate
236215 SCRIPT_FOLDER=oss_scripts
237- # 16bit models will encounter op validation fail on some operations,
216+ # Bert models running in 16bit will encounter op validation fail on some operations,
238217 # which requires CHIPSET >= SM8550.
239218 QNN_CHIPSET=SM8550
240219 ;;
0 commit comments