From 9af7a5a62c936c1c7a036c0589f09abcb118db74 Mon Sep 17 00:00:00 2001 From: Mergen Nachin Date: Wed, 15 Jan 2025 10:58:51 -0500 Subject: [PATCH] Update torchtune pin (#7670) (cherry picked from commit a18f6e89631ab97f6211aaae537bfe2cdd90729e) --- examples/models/llama3_2_vision/install_requirements.sh | 7 ++----- 1 file changed, 2 insertions(+), 5 deletions(-) diff --git a/examples/models/llama3_2_vision/install_requirements.sh b/examples/models/llama3_2_vision/install_requirements.sh index de8ed0632d7..4d4a6f28624 100755 --- a/examples/models/llama3_2_vision/install_requirements.sh +++ b/examples/models/llama3_2_vision/install_requirements.sh @@ -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