diff --git a/gpt2-got/gpt2-model-training/requirements-local.txt b/gpt2-got/gpt2-model-training/requirements-local.txt
index 49e214a..09d733c 100644
--- a/gpt2-got/gpt2-model-training/requirements-local.txt
+++ b/gpt2-got/gpt2-model-training/requirements-local.txt
@@ -1,2 +1,2 @@
-tensorflow==1.15.2
+tensorflow==2.6.4
 gpt-2-simple
\ No newline at end of file