modify requirements

2022-11-04 07:10:55 +00:00 · 2022-11-04 07:10:55 +00:00 · b0e7e5b619
parent 3307f4061c
commit b0e7e5b619
2 changed files with 3 additions and 3 deletions
--- 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
--- 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