diff --git a/setup.py b/setup.py
index 61cbf80..7307372 100644
--- a/setup.py
+++ b/setup.py
@@ -8,10 +8,10 @@
 
 readme = open('README.md').read()
 
-VERSION = '0.0.31'
+VERSION = '0.0.4'
 
 requirements = [
-    'torch>=1.0.0',
+    'torch',
 ]
 
 # import subprocess