Skip to content

Commit 00305f0

Browse files
mergennachinper
authored andcommitted
Update torchtune pin (pytorch#7670)
(cherry picked from commit a18f6e8)
1 parent 0431a79 commit 00305f0

File tree

1 file changed

+2
-5
lines changed

1 file changed

+2
-5
lines changed

examples/models/llama3_2_vision/install_requirements.sh

+2-5
Original file line numberDiff line numberDiff line change
@@ -5,10 +5,7 @@
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"
8+
NIGHTLY_VERSION="dev20250115"
99

1010
# Install torchtune nightly for model definitions.
11-
pip install --pre torchtune==0.4.0.${NIGHTLY_VERSION} --extra-index-url https://download.pytorch.org/whl/nightly/cpu --no-cache-dir
12-
13-
# Install torchao.
14-
pip install "$(dirname "$0")/../../../third-party/ao"
11+
pip install --pre torchtune==0.6.0.${NIGHTLY_VERSION} --extra-index-url https://download.pytorch.org/whl/nightly/cpu --no-cache-dir

0 commit comments

Comments
 (0)