Skip to content
This repository was archived by the owner on Feb 3, 2025. It is now read-only.

Commit 7a40ba9

Browse files
author
DEKHTIARJonathan
committed
[Benchmarking-Py] Bug Fix - T5 Dependency was overwritting TF
1 parent dbbfd33 commit 7a40ba9

File tree

1 file changed

+3
-1
lines changed

1 file changed

+3
-1
lines changed

tftrt/benchmarking-python/huggingface/t5/base_run_inference.sh

Lines changed: 3 additions & 1 deletion
Original file line numberDiff line numberDiff line change
@@ -105,10 +105,12 @@ fi
105105

106106
# Install Dependencies
107107

108+
TF_VERSION=$(python -c "import tensorflow as tf; print('.'.join(tf.__version__.split('.')[:2]))")
108109
pip install --upgrade \
109110
prefetch_generator \
110111
orjson \
111-
t5==0.4.0
112+
t5==0.4.0 \
113+
tensorflow-text==${TF_VERSION}
112114

113115
set -x
114116

0 commit comments

Comments
 (0)