Skip to content

Commit ee14ad0

Browse files
authored
Pin TorchTune (#6791)
1 parent 4f2ac7e commit ee14ad0

File tree

1 file changed

+3
-1
lines changed

1 file changed

+3
-1
lines changed

examples/models/llama3_2_vision/install_requirements.sh

Lines changed: 3 additions & 1 deletion
Original file line numberDiff line numberDiff line change
@@ -5,8 +5,10 @@
55
# This source code is licensed under the BSD-style license found in the
66
# LICENSE file in the root directory of this source tree.
77

8+
NIGHTLY_VERSION="dev20241112"
9+
810
# Install torchtune nightly for model definitions.
9-
pip install --pre torchtune --extra-index-url https://download.pytorch.org/whl/nightly/cpu --no-cache-dir
11+
pip install --pre torchtune==0.4.0.${NIGHTLY_VERSION} --extra-index-url https://download.pytorch.org/whl/nightly/cpu --no-cache-dir
1012

1113
# Install torchao.
1214
pip install "$(dirname "$0")/../../../third-party/ao"

0 commit comments

Comments
 (0)