diff --git a/requirements.txt b/requirements.txt
index 12251a9..a4d7b04 100644
--- a/requirements.txt
+++ b/requirements.txt
@@ -1,6 +1,6 @@
 torch>=1.8.0
 transformers>=4.10.0
-datasets==1.17.0
+datasets>=1.17.0
 sentencepiece>=0.1.96
 tqdm>=4.62.2
 decorator
diff --git a/setup.py b/setup.py
index 711f5a4..2324fcd 100644
--- a/setup.py
+++ b/setup.py
@@ -6,7 +6,7 @@ import os
 
 requires = """torch>=1.8.0
 transformers>=4.10.0
-datasets==1.17.0
+datasets>=1.17.0
 sentencepiece>=0.1.96
 tqdm>=4.62.2
 decorator
@@ -58,4 +58,4 @@ with open('README.md', 'r') as f:
             "License :: OSI Approved :: Apache Software License",
             "Operating System :: OS Independent",
         ]
-    )
\ No newline at end of file
+    )