File tree Expand file tree Collapse file tree 4 files changed +13
-1
lines changed
.ci/docker/ci_commit_pins Expand file tree Collapse file tree 4 files changed +13
-1
lines changed Original file line number Diff line number Diff line change
1
+ 0916b5b29b092afcbf2b898caae49abe80662bac
Original file line number Diff line number Diff line change 7
7
8
8
# Install torchtune nightly for model definitions.
9
9
pip install --pre torchtune --extra-index-url https://download.pytorch.org/whl/nightly/cpu --no-cache-dir
10
+
11
+ # Install torchao.
12
+ TORCHAO_VERSION=$( cat " $( dirname " $0 " ) " /../../../.ci/docker/ci_commit_pins/torchao.txt)
13
+ pip install --no-use-pep517 --user " git+https://github.com/pytorch/ao.git@${TORCHAO_VERSION} "
Original file line number Diff line number Diff line change 8
8
# Install snakeviz for cProfile flamegraph
9
9
# Install sentencepiece for llama tokenizer
10
10
pip install snakeviz sentencepiece
11
- pip install torchao==0.1
11
+
12
+ # Install torchao.
13
+ TORCHAO_VERSION=$( cat " $( dirname " $0 " ) " /../../../.ci/docker/ci_commit_pins/torchao.txt)
14
+ pip install --no-use-pep517 --user " git+https://github.com/pytorch/ao.git@${TORCHAO_VERSION} "
12
15
13
16
# Install lm-eval for Model Evaluation with lm-evalution-harness
14
17
# Install tiktoken for tokenizer
Original file line number Diff line number Diff line change 8
8
pip install torchvision
9
9
pip install torchtune
10
10
pip install tiktoken
11
+
12
+ # Install torchao.
13
+ TORCHAO_VERSION=$( cat " $( dirname " $0 " ) " /../../../.ci/docker/ci_commit_pins/torchao.txt)
14
+ pip install --no-use-pep517 --user " git+https://github.com/pytorch/ao.git@${TORCHAO_VERSION} "
You can’t perform that action at this time.
0 commit comments