Skip to content

Commit a0fbe00

Browse files
committed
Update on "Add preprocess to ci"
1. Run `test_preprocess.py` in ci. 2. Refactor `test_preprocess.py` by creating the model artifacts (eager, exported, et) once and re-use them on different inputs (previously, create model artifacts for each test). Test Plan: ``` pytest -c /dev/null -v -n auto examples/models/llama3_2_vision/preprocess/ ``` [ghstack-poisoned]
2 parents 62fe73b + 607ea6f commit a0fbe00

File tree

2 files changed

+4
-1
lines changed

2 files changed

+4
-1
lines changed

.github/workflows/_unittest.yml

Lines changed: 3 additions & 1 deletion
Original file line numberDiff line numberDiff line change
@@ -38,7 +38,7 @@ jobs:
3838
.ci/scripts/setup-linux.sh cmake
3939
4040
# Install llama3_2_vision dependencies.
41-
bash examples/models/llama3_2_vision/install_requirements.sh
41+
PYTHON_EXECUTABLE=python ./examples/models/llama3_2_vision/install_requirements.sh
4242
4343
# Run pytest with coverage
4444
pytest -n auto --cov=./ --cov-report=xml
@@ -70,6 +70,8 @@ jobs:
7070
${CONDA_RUN} --no-capture-output \
7171
.ci/scripts/setup-macos.sh cmake
7272
73+
${PYTHON_EXECUTABLE} pip list
74+
7375
# Install llama3_2_vision dependencies.
7476
bash examples/models/llama3_2_vision/install_requirements.sh
7577

examples/models/llama3_2_vision/install_requirements.sh

Lines changed: 1 addition & 0 deletions
Original file line numberDiff line numberDiff line change
@@ -8,6 +8,7 @@
88
# Install torchtune nightly for model definitions.
99
echo "lfq check pip"
1010
pip list # debugging
11+
echo "lfq end pip"
1112
pip install --pre torchtune --extra-index-url https://download.pytorch.org/whl/nightly/cpu --no-cache-dir
1213

1314
# Install torchao.

0 commit comments

Comments
 (0)