diff --git a/requirements.txt b/requirements.txt index 6c8484f..6886de6 100644 --- a/requirements.txt +++ b/requirements.txt @@ -89,7 +89,7 @@ t5==0.8.0 tensor2tensor==1.15.7 tensorboard==2.2.2 tensorboard-plugin-wit==1.6.0.post3 -tensorflow==2.5.0 +tensorflow==2.7.2 tensorflow-addons==0.10.0 tensorflow-datasets==3.1.0 tensorflow-estimator==2.2.0 diff --git a/setup.py b/setup.py index 556d28f..b09e26f 100644 --- a/setup.py +++ b/setup.py @@ -30,7 +30,7 @@ 'tensor2tensor' ], extras_require={ - 'tensorflow': ['tensorflow==2.5.0'], + 'tensorflow': ['tensorflow==2.7.2'], 'tensorflow_gpu': ['tensorflow-gpu==2.2.2'], 'tensorflow-hub': ['tensorflow-hub>=0.6.0'], },