Skip to content

Commit

Permalink
Revert keras version to 3.3.3 in build system
Browse files Browse the repository at this point in the history
  • Loading branch information
laxmareddyp committed Sep 18, 2024
1 parent cba86c2 commit 621cdc1
Showing 1 changed file with 3 additions and 3 deletions.
6 changes: 3 additions & 3 deletions .kokoro/github/ubuntu/gpu/build.sh
Original file line number Diff line number Diff line change
Expand Up @@ -36,23 +36,23 @@ then
pip install -r requirements-tensorflow-cuda.txt --progress-bar off --timeout 1000
pip install keras-nlp-nightly --no-deps
pip install tensorflow-text~=2.16.0
pip install keras~=3.5.0
pip install keras~=3.3.3

elif [ "$KERAS_BACKEND" == "jax" ]
then
echo "JAX backend detected."
pip install -r requirements-jax-cuda.txt --progress-bar off --timeout 1000
pip install keras-nlp-nightly --no-deps
pip install tensorflow-text~=2.16.0
pip install keras~=3.5.0
pip install keras~=3.3.3

elif [ "$KERAS_BACKEND" == "torch" ]
then
echo "PyTorch backend detected."
pip install -r requirements-torch-cuda.txt --progress-bar off --timeout 1000
pip install keras-nlp-nightly --no-deps
pip install tensorflow-text~=2.16.0
pip install keras~=3.5.0
pip install keras~=3.3.3
fi

pip install --no-deps -e "." --progress-bar off
Expand Down

0 comments on commit 621cdc1

Please sign in to comment.