From 72bf30fcd4f77ccee6678e33b19995b341385eb0 Mon Sep 17 00:00:00 2001 From: Jack Zhang Date: Tue, 12 Nov 2024 11:59:14 -0800 Subject: [PATCH 1/2] Add TorchTune pin --- examples/models/llama3_2_vision/install_requirements.sh | 4 +++- 1 file changed, 3 insertions(+), 1 deletion(-) diff --git a/examples/models/llama3_2_vision/install_requirements.sh b/examples/models/llama3_2_vision/install_requirements.sh index 49558952d87..3dd8e6d9f79 100755 --- a/examples/models/llama3_2_vision/install_requirements.sh +++ b/examples/models/llama3_2_vision/install_requirements.sh @@ -5,8 +5,10 @@ # 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="dev20241106" + # Install torchtune nightly for model definitions. -pip install --pre torchtune --extra-index-url https://download.pytorch.org/whl/nightly/cpu --no-cache-dir +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" From f24334a1f893be47748d9402df1b317145c82ff2 Mon Sep 17 00:00:00 2001 From: Jack Zhang Date: Wed, 13 Nov 2024 07:45:19 -0800 Subject: [PATCH 2/2] Bump to match recent InferenceKVCache changes --- examples/models/llama3_2_vision/install_requirements.sh | 2 +- 1 file changed, 1 insertion(+), 1 deletion(-) diff --git a/examples/models/llama3_2_vision/install_requirements.sh b/examples/models/llama3_2_vision/install_requirements.sh index 3dd8e6d9f79..de8ed0632d7 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="dev20241106" +NIGHTLY_VERSION="dev20241112" # 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