temp

Author: winskuo-quic

.ci/scripts/test_model.sh

@@ -167,10 +167,12 @@ test_model_with_qnn() {
   elif [[ "${MODEL_NAME}" == "vit" ]]; then
     EXPORT_SCRIPT=torchvision_vit
   elif [[ "${MODEL_NAME}" == "mb" ]]; then
-    pip install sklearn
+    pip install scikit-learn
     EXPORT_SCRIPT=mobilebert_fine_tune
   elif [[ "${MODEL_NAME}" == "w2l" ]]; then
     EXPORT_SCRIPT=wav2letter
+    echo "W2L Checking for Version"
+    python -c "import torchaudio; print(torchaudio.__version__)"
   elif [[ "${MODEL_NAME}" == "edsr" ]]; then
     EXPORT_SCRIPT=edsr
     # Additional deps for edsr