uninstall tbp-nightly ahead of time

AI-Hypercomputer · Oct 4, 2023 · 318e852 · 318e852
1 parent 87e5769
commit 318e852
Showing 1 changed file with 2 additions and 1 deletion.
diff --git a/setup.sh b/setup.sh
@@ -66,10 +66,11 @@ fi
 # Save the script folder path of maxtext
 run_name_folder_path=$(pwd)
 
-# Uninstall existing jax, jaxlib, and libtpu-nightly
+# Uninstall existing jax, jaxlib, libtpu-nightly and tbp-nightly
 pip3 show jax && pip3 uninstall -y jax 
 pip3 show jaxlib && pip3 uninstall -y jaxlib
 pip3 show libtpu-nightly && pip3 uninstall -y libtpu-nightly
+pip3 show tbp-nightly && pip3 uninstall -y tbp-nightly
 
 # Delete jax folder if it exists
 if [[ -d $HOME/jax ]]; then