diff --git a/albert/requirements.txt b/albert/requirements.txt index 65c9197..6fad652 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.11.1 # CPU Version of TensorFlow tensorflow_hub==0.7 # tensorflow-gpu==1.15 # GPU version of TensorFlow sentencepiece