debug

Author: cccclai

.github/workflows/trunk.yml

@@ -828,25 +828,27 @@ jobs:
         CONDA_ENV=$(conda env list --json | jq -r ".envs | .[-1]")
         conda activate "${CONDA_ENV}"
 
+
+        # Install executorch
+        echo "::group::Set up ExecuTorch"
+        ./install_requirements.sh --use-pt-pinned-commit
+        echo "::endgroup::"
+
         # Used by .ci/scripts/test_huggingface_optimum_model.py
         pip install datasets
 
         echo "::group::Set up Hugging Face"
-        pip install -U "huggingface_hub[cli]"
+        pip install -U "huggingface_hub[cli]" accelerate
         huggingface-cli login --token $SECRET_EXECUTORCH_HF_TOKEN
         OPTIMUM_ET_VERSION=8335b491fa77a6c43709b4dfc192184add6ef224
         pip install git+https://github.com/huggingface/optimum-executorch.git@${OPTIMUM_ET_VERSION}
         echo "::endgroup::"
 
+        ${CONDA_RUN} python -c "import torch, torchvision; print(torch.__version__, torchvision.__version__)"
+
         echo "::group::Run tests"
         export OUTPUT_DIR="$(pwd)/${MODEL}_${RECIPE}_${QUANTIZE}"
-        python .ci/scripts/test_huggingface_optimum_model.py --model ${MODEL} --recipe ${RECIPE} ${QUANTIZE} --model_dir ${OUTPUT_DIR}
-        echo "::endgroup::"
-
-        # Install executorch
-        echo "::group::Set up ExecuTorch"
-        ./install_requirements.sh --use-pt-pinned-commit
-
+        ${CONDA_RUN} python .ci/scripts/test_huggingface_optimum_model.py --model ${MODEL} --recipe ${RECIPE} ${QUANTIZE} --model_dir ${OUTPUT_DIR}
         echo "::endgroup::"
 
         echo "::group::Generate artifacts for performance profiling"