Skip to content

Commit

Permalink
Update torchtune pin (pytorch#7670)
Browse files Browse the repository at this point in the history
(cherry picked from commit a18f6e8)
  • Loading branch information
mergennachin authored and per committed Jan 16, 2025
1 parent 0431a79 commit 00305f0
Showing 1 changed file with 2 additions and 5 deletions.
7 changes: 2 additions & 5 deletions examples/models/llama3_2_vision/install_requirements.sh
Original file line number Diff line number Diff line change
Expand Up @@ -5,10 +5,7 @@
# This source code is licensed under the BSD-style license found in the
# LICENSE file in the root directory of this source tree.

NIGHTLY_VERSION="dev20241112"
NIGHTLY_VERSION="dev20250115"

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

# Install torchao.
pip install "$(dirname "$0")/../../../third-party/ao"
pip install --pre torchtune==0.6.0.${NIGHTLY_VERSION} --extra-index-url https://download.pytorch.org/whl/nightly/cpu --no-cache-dir

0 comments on commit 00305f0

Please sign in to comment.