remove pt update

metascroy · metascroy · commit f962c5169a9f · 2024-09-30T11:58:37.000-07:00
diff --git a/install/install_requirements.sh b/install/install_requirements.sh
@@ -91,12 +91,6 @@ 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
@@ -107,5 +101,5 @@ fi
 
 (
   set -x
-  $PIP_EXECUTABLE install lm-eval=="0.4.2"
+  $PIP_EXECUTABLE install evaluate=="0.4.3" lm-eval=="0.4.2" psutil=="6.0.0"
 )

Original file line number	Diff line number	Diff line change
`@@ -91,12 +91,6 @@ REQUIREMENTS_TO_INSTALL=(`
`91`	`91`	`$PIP_EXECUTABLE install torchao=="0.5.0"`
`92`	`92`	`)`
`93`	`93`
`94`		`-# Rely on the latest tochtune for flamingo support`
`95`		`-(`
`96`		`- set -x`
`97`		`- $PIP_EXECUTABLE install -I git+https://github.com/pytorch/torchtune.git@d002d45e3ec700fa770d9dcc61b02c59e2507bf6`
`98`		`-)`
`99`		`-`
`100`	`94`	`if [[ -x "$(command -v nvidia-smi)" ]]; then`
`101`	`95`	`(`
`102`	`96`	`set -x`
`@@ -107,5 +101,5 @@ fi`
`107`	`101`
`108`	`102`	`(`
`109`	`103`	`set -x`
`110`		`- $PIP_EXECUTABLE install lm-eval=="0.4.2"`
	`104`	`+ $PIP_EXECUTABLE install evaluate=="0.4.3" lm-eval=="0.4.2" psutil=="6.0.0"`
`111`	`105`	`)`