remove pt update

Author: metascroy

install/install_requirements.sh

@@ -47,13 +47,10 @@ fi
 # NOTE: If a newly-fetched version of the executorch repo changes the value of
 # PYTORCH_NIGHTLY_VERSION, you should re-run this script to install the necessary
 # package versions.
-PYTORCH_NIGHTLY_VERSION=dev20240925
+PYTORCH_NIGHTLY_VERSION=dev20240901
 
 # Nightly version for torchvision
-VISION_NIGHTLY_VERSION=dev20240925
-
-# Nightly version for torchtune
-TUNE_NIGHTLY_VERSION=dev20240928
+VISION_NIGHTLY_VERSION=dev20240901
 
 # Uninstall triton, as nightly will depend on pytorch-triton, which is one and the same
 (
@@ -73,9 +70,8 @@ fi
 
 # pip packages needed by exir.
 REQUIREMENTS_TO_INSTALL=(
-  torch=="2.6.0.${PYTORCH_NIGHTLY_VERSION}"
+  torch=="2.5.0.${PYTORCH_NIGHTLY_VERSION}"
   torchvision=="0.20.0.${VISION_NIGHTLY_VERSION}"
-  torchtune=="0.3.0.${TUNE_NIGHTLY_VERSION}"
 )
 
 # Install the requirements. --extra-index-url tells pip to look for package
@@ -91,6 +87,12 @@ REQUIREMENTS_TO_INSTALL=(
   $PIP_EXECUTABLE install torchao=="0.5.0"
 )
 
+# Rely on the latest tochtune for flamingo support
+(
+  set -x
+  $PIP_EXECUTABLE install -I git+https://github.com/pytorch/torchtune.git@d002d45e3ec700fa770d9dcc61b02c59e2507bf6
+)
+
 if [[ -x "$(command -v nvidia-smi)" ]]; then
   (
     set -x