Skip to content

Commit

Permalink
Places a constraint on tensorflow-text when installing t5x
Browse files Browse the repository at this point in the history
This change just excludes TF 2.13.0 which has this issue, but looking
at the head of tensorflow, it looks like the constraint on the
problematic package (typing-extensions) was relaxed
  • Loading branch information
terrykong committed Jul 12, 2023
1 parent 50a0197 commit 637fd2c
Showing 1 changed file with 7 additions and 1 deletion.
8 changes: 7 additions & 1 deletion .github/container/install-t5x.sh
Original file line number Diff line number Diff line change
Expand Up @@ -70,7 +70,13 @@ apt-get install -y \
git clone ${T5X_REPO} ${INSTALL_DIR}/t5x
cd ${INSTALL_DIR}/t5x
git checkout ${T5X_REF}
pip install -e .[gpu]
# The dependency chain for t5x is t5x -> seqio[any] -> tensorflow-text[any] -> tensorflow[any]
# and tensorflow 2.13.0 has a constraint on typing_extensions that makes it
# incompatible with transformer_engine which depends on pydantic and by
# extension, typing_extensions: https://github.com/tensorflow/tensorflow/pull/60688
# As the top level application, we will just exclude it since typing_extensions appears
# to have been relaxed at the head of tensorflow.
pip install -e .[gpu] 'tensorflow-text!=2.13.*'

apt-get autoremove -y
apt-get clean
Expand Down

0 comments on commit 637fd2c

Please sign in to comment.