diff --git a/albert/requirements.txt b/albert/requirements.txt index 65c9197..5e47beb 100644 --- a/albert/requirements.txt +++ b/albert/requirements.txt @@ -1,5 +1,5 @@ # Run pip install --upgrade pip if tensorflow 1.15 cannot be found -tensorflow==1.15.2 # CPU Version of TensorFlow +tensorflow==2.5.1 # CPU Version of TensorFlow tensorflow_hub==0.7 # tensorflow-gpu==1.15 # GPU version of TensorFlow sentencepiece diff --git a/requirements.txt b/requirements.txt index 9e41fae..e2ceec7 100644 --- a/requirements.txt +++ b/requirements.txt @@ -1,5 +1,5 @@ # Run pip install --upgrade pip if tensorflow 1.15 cannot be found -tensorflow==1.15.2 # CPU Version of TensorFlow +tensorflow==2.5.1 # CPU Version of TensorFlow # tensorflow-gpu==1.15 # GPU version of TensorFlow tensorflow_hub==0.7 torch==1.6.0 diff --git a/requirements_colab.txt b/requirements_colab.txt index 1bde112..fe23d07 100644 --- a/requirements_colab.txt +++ b/requirements_colab.txt @@ -1,5 +1,5 @@ # Run pip install --upgrade pip if tensorflow 1.15 cannot be found -tensorflow==1.15.2 # CPU Version of TensorFlow +tensorflow==2.5.1 # CPU Version of TensorFlow # tensorflow-gpu==1.15 # GPU version of TensorFlow tensorflow_hub==0.7 # torch==1.6.0