From 621cdc11c8b39520d60977fbf67682bff5b02704 Mon Sep 17 00:00:00 2001 From: Laxma Reddy Patlolla Date: Wed, 18 Sep 2024 13:55:20 -0700 Subject: [PATCH] Revert keras version to 3.3.3 in build system --- .kokoro/github/ubuntu/gpu/build.sh | 6 +++--- 1 file changed, 3 insertions(+), 3 deletions(-) diff --git a/.kokoro/github/ubuntu/gpu/build.sh b/.kokoro/github/ubuntu/gpu/build.sh index 922922fcb0..5c71fe0931 100644 --- a/.kokoro/github/ubuntu/gpu/build.sh +++ b/.kokoro/github/ubuntu/gpu/build.sh @@ -36,7 +36,7 @@ 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 @@ -44,7 +44,7 @@ then 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 @@ -52,7 +52,7 @@ then 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