Skip to content

Commit af07932

Browse files
authored
Pin tf-text to 2.18.1 and remove default branch from internal repos (#1387)
1. Similar to maxtext, we need to pin `tensorflow==2.18.1` 2. Remove default branch `main` for any internal repos
1 parent 25afc1d commit af07932

File tree

2 files changed

+4
-0
lines changed

2 files changed

+4
-0
lines changed

.github/container/Dockerfile.t5x

Lines changed: 3 additions & 0 deletions
Original file line numberDiff line numberDiff line change
@@ -24,6 +24,9 @@ echo "-e file://${SRC_PATH_T5X}[gpu]" >> /opt/pip-tools.d/requirements-t5x.in
2424
# so setting the lower-bound to something recent
2525
echo "seqio-nightly>=0.0.18.dev20240714" >> /opt/pip-tools.d/requirements-t5x.in
2626

27+
# tensorflow-cpu==2.19.0 is incompatible with tensorflow-text
28+
echo "tensorflow==2.18.1" >> /opt/pip-tools.d/requirements-t5x.in
29+
2730
# 2. Remove head-of-tree specs from select dependencies
2831
pushd ${SRC_PATH_T5X}
2932
sed -i "s| @ git+https://github.com/google/flax#egg=flax||g" setup.py

.github/container/git-clone.sh

Lines changed: 1 addition & 0 deletions
Original file line numberDiff line numberDiff line change
@@ -82,6 +82,7 @@ if [[ "${GIT_REPO}" == *"gitlab"* ]]; then
8282
if grep -q -r gitlab-ci-token .git; then
8383
grep -r gitlab-ci-token .git | awk -F: '{print $1}' | xargs rm -f
8484
fi
85+
git branch -D main
8586
fi
8687
popd
8788

0 commit comments

Comments
 (0)