diff --git a/ternarybert/requirements.txt b/ternarybert/requirements.txt
index 7bbcdf0a..c4e5fc98 100644
--- a/ternarybert/requirements.txt
+++ b/ternarybert/requirements.txt
@@ -3,5 +3,5 @@ requests
 scipy
 future
 Pillow
-tensorflow==1.14.0
+tensorflow==2.4.0
 torch==1.1.0