Skip to content

Commit d8593fd

Browse files
committed
uninstall torch/torchvision etc
1 parent d3d53c1 commit d8593fd

File tree

1 file changed

+2
-1
lines changed

1 file changed

+2
-1
lines changed

.github/workflows/trunk.yml

Lines changed: 2 additions & 1 deletion
Original file line numberDiff line numberDiff line change
@@ -828,6 +828,7 @@ jobs:
828828
CONDA_ENV=$(conda env list --json | jq -r ".envs | .[-1]")
829829
conda activate "${CONDA_ENV}"
830830
831+
pip uninstall torch torchaudio torchvision -y
831832
832833
# Install executorch
833834
echo "::group::Set up ExecuTorch"
@@ -848,7 +849,7 @@ jobs:
848849
849850
echo "::group::Run tests"
850851
export OUTPUT_DIR="$(pwd)/${MODEL}_${RECIPE}_${QUANTIZE}"
851-
${CONDA_RUN} python .ci/scripts/test_huggingface_optimum_model.py --model ${MODEL} --recipe ${RECIPE} ${QUANTIZE} --model_dir ${OUTPUT_DIR}
852+
python .ci/scripts/test_huggingface_optimum_model.py --model ${MODEL} --recipe ${RECIPE} ${QUANTIZE} --model_dir ${OUTPUT_DIR}
852853
echo "::endgroup::"
853854
854855
echo "::group::Generate artifacts for performance profiling"

0 commit comments

Comments
 (0)