diff --git a/install_everything.sh b/install_everything.sh index 2d1c0929..1dd32ae2 100644 --- a/install_everything.sh +++ b/install_everything.sh @@ -3,7 +3,7 @@ TORCHXLA_TAG=jetstream-pytorch pip3 install pip install jax[tpu] -f https://storage.googleapis.com/jax-releases/libtpu_releases.html # torch cpu pip3 install torch --index-url https://download.pytorch.org/whl/cpu -pip3 install tensorflow flatbuffers absl-py flax sentencepiece seqio google-cloud-storage safetensors +pip3 install tensorflow flatbuffers absl-py flax sentencepiece seqio google-cloud-storage safetensors colorama mkdir -p deps pushd deps