diff --git a/requirements.txt b/requirements.txt
index 6488ec2..3f0c464 100644
--- a/requirements.txt
+++ b/requirements.txt
@@ -16,7 +16,7 @@ TatSu==4.4.0
 textworld==1.2.0
 
 # NN required
-tensorflow==1.15.4
+tensorflow==2.4.0
 bert-tensorflow==1.0.1
 albert-tensorflow==1.1
 sentencepiece==0.1.85