Skip to content

Commit 1830375

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 a0fbe00 + 16ea55a commit 1830375

File tree

1 file changed

+0
-4
lines changed

1 file changed

+0
-4
lines changed

examples/models/llama3_2_vision/install_requirements.sh

Lines changed: 0 additions & 4 deletions
Original file line numberDiff line numberDiff line change
@@ -6,12 +6,8 @@
66
# LICENSE file in the root directory of this source tree.
77

88
# Install torchtune nightly for model definitions.
9-
echo "lfq check pip"
10-
pip list # debugging
11-
echo "lfq end pip"
129
pip install --pre torchtune --extra-index-url https://download.pytorch.org/whl/nightly/cpu --no-cache-dir
1310

1411
# Install torchao.
15-
pip list # debugging
1612
TORCHAO_VERSION=$(cat "$(dirname "$0")"/../../../.ci/docker/ci_commit_pins/torchao.txt)
1713
pip install --no-use-pep517 "git+https://github.com/pytorch/ao.git@${TORCHAO_VERSION}"

0 commit comments

Comments
 (0)