diff --git a/requirements.txt b/requirements.txt
index 94a636c..97025ba 100644
--- a/requirements.txt
+++ b/requirements.txt
@@ -2,7 +2,7 @@ pytorch-transformers==1.0.0
 numpy==1.16.4
 lmdb==0.94
 tensorboardX==1.2
-tensorflow==2.5.0
+tensorflow
 tensorpack==0.9.4
 tqdm==4.31.1
 easydict==1.9