Skip to content

Commit 52c9f30

Browse files
authored
[executorch] Sync torchao version
Differential Revision: D61925882 Pull Request resolved: #4945
1 parent a5157de commit 52c9f30

File tree

4 files changed

+13
-1
lines changed

4 files changed

+13
-1
lines changed
Lines changed: 1 addition & 0 deletions
Original file line numberDiff line numberDiff line change
@@ -0,0 +1 @@
1+
0916b5b29b092afcbf2b898caae49abe80662bac

examples/models/flamingo/install_requirements.sh

Lines changed: 4 additions & 0 deletions
Original file line numberDiff line numberDiff line change
@@ -7,3 +7,7 @@
77

88
# Install torchtune nightly for model definitions.
99
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}"

examples/models/llama2/install_requirements.sh

Lines changed: 4 additions & 1 deletion
Original file line numberDiff line numberDiff line change
@@ -8,7 +8,10 @@
88
# Install snakeviz for cProfile flamegraph
99
# Install sentencepiece for llama tokenizer
1010
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}"
1215

1316
# Install lm-eval for Model Evaluation with lm-evalution-harness
1417
# Install tiktoken for tokenizer

examples/models/phi-3-mini-lora/install_requirements.sh

Lines changed: 4 additions & 0 deletions
Original file line numberDiff line numberDiff line change
@@ -8,3 +8,7 @@
88
pip install torchvision
99
pip install torchtune
1010
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}"

0 commit comments

Comments
 (0)