diff --git a/requirements.txt b/requirements.txt
index ec974194..5c69cd6e 100644
--- a/requirements.txt
+++ b/requirements.txt
@@ -1,7 +1,7 @@
 transformers == 4.5.1
-tensorflow-gpu == 2.4.1
+tensorflow-gpu
 Flask == 1.1.2
 Flask-SocketIO == 5.0.1
 requests == 2.25.1
 easygui == 0.98.2
-torch == 1.8.1+cu111
\ No newline at end of file
+torch == 1.8.1
\ No newline at end of file