From d532a9c35b62262aa645c24d40e9159d5f594068 Mon Sep 17 00:00:00 2001 From: bachvudinh Date: Thu, 18 Jul 2024 16:48:39 +0700 Subject: [PATCH] update dependency --- setup.py | 4 ++-- 1 file changed, 2 insertions(+), 2 deletions(-) diff --git a/setup.py b/setup.py index 8c3e35e..222ef77 100644 --- 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',