From 480356f73275cc79c8fbbadad9d3bde09d069a8c Mon Sep 17 00:00:00 2001 From: Mergen Nachin Date: Wed, 15 Jan 2025 10:31:53 -0500 Subject: [PATCH] Update torchtune pin --- examples/models/llama3_2_vision/install_requirements.sh | 4 ++-- 1 file changed, 2 insertions(+), 2 deletions(-) diff --git a/examples/models/llama3_2_vision/install_requirements.sh b/examples/models/llama3_2_vision/install_requirements.sh index a4d789d56e..4d4a6f2862 100755 --- a/examples/models/llama3_2_vision/install_requirements.sh +++ b/examples/models/llama3_2_vision/install_requirements.sh @@ -5,7 +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 +pip install --pre torchtune==0.6.0.${NIGHTLY_VERSION} --extra-index-url https://download.pytorch.org/whl/nightly/cpu --no-cache-dir