update dependency

janhq · Jul 18, 2024 · d532a9c · d532a9c
1 parent 11d6460
commit d532a9c
Showing 1 changed file with 2 additions and 2 deletions.
diff --git a/setup.py b/setup.py
@@ -13,10 +13,10 @@
     install_requires=[
         'boto3',
         # tokenizers >=0.13.3
-        'torch',
         'tokenizers==0.13.3',
         'transformers',
-        'datasets==2.20.0',  # Add the datasets library
+        'datasets==2.20.0', 
+        'torch', # Add the datasets library
     ],
     classifiers=[
         'Programming Language :: Python :: 3',