remove pt update

metascroy · metascroy · commit ff3f2fddadca · 2024-09-30T11:57:08.000-07:00
diff --git a/install/install_requirements.sh b/install/install_requirements.sh
@@ -52,6 +52,9 @@ PYTORCH_NIGHTLY_VERSION=dev20240901
 # Nightly version for torchvision
 VISION_NIGHTLY_VERSION=dev20240901
 
+# Nightly version for torchtune
+TUNE_NIGHTLY_VERSION=dev20240928
+
 # Uninstall triton, as nightly will depend on pytorch-triton, which is one and the same
 (
   set -x
@@ -72,6 +75,7 @@ fi
 REQUIREMENTS_TO_INSTALL=(
   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
@@ -103,5 +107,5 @@ fi
 
 (
   set -x
-  $PIP_EXECUTABLE install evaluate=="0.4.3" lm-eval=="0.4.2" psutil=="6.0.0"
+  $PIP_EXECUTABLE install lm-eval=="0.4.2"
 )

Original file line number	Diff line number	Diff line change
`@@ -52,6 +52,9 @@ PYTORCH_NIGHTLY_VERSION=dev20240901`
`52`	`52`	`# Nightly version for torchvision`
`53`	`53`	`VISION_NIGHTLY_VERSION=dev20240901`
`54`	`54`
	`55`	`+# Nightly version for torchtune`
	`56`	`+TUNE_NIGHTLY_VERSION=dev20240928`
	`57`	`+`
`55`	`58`	`# Uninstall triton, as nightly will depend on pytorch-triton, which is one and the same`
`56`	`59`	`(`
`57`	`60`	`set -x`
`@@ -72,6 +75,7 @@ fi`
`72`	`75`	`REQUIREMENTS_TO_INSTALL=(`
`73`	`76`	`torch=="2.5.0.${PYTORCH_NIGHTLY_VERSION}"`
`74`	`77`	`torchvision=="0.20.0.${VISION_NIGHTLY_VERSION}"`
	`78`	`+ torchtune=="0.3.0.${TUNE_NIGHTLY_VERSION}"`
`75`	`79`	`)`
`76`	`80`
`77`	`81`	`# Install the requirements. --extra-index-url tells pip to look for package`
`@@ -103,5 +107,5 @@ fi`
`103`	`107`
`104`	`108`	`(`
`105`	`109`	`set -x`
`106`		`- $PIP_EXECUTABLE install evaluate=="0.4.3" lm-eval=="0.4.2" psutil=="6.0.0"`
	`110`	`+ $PIP_EXECUTABLE install lm-eval=="0.4.2"`
`107`	`111`	`)`