diff --git a/requirements.txt b/requirements.txt index aa2ffcd..37ab607 100644 --- a/requirements.txt +++ b/requirements.txt @@ -8,4 +8,4 @@ sklearn==0.0 tabulate==0.8.6 torch==1.4.0 tqdm==4.42.1 -transformers==2.4.1 +transformers==4.30.0 diff --git a/setup.py b/setup.py index 8ceaf55..8b08bd0 100644 --- a/setup.py +++ b/setup.py @@ -10,7 +10,7 @@ description='文本分类工具集', packages=find_packages(), install_requires=[ - "transformers==2.4.1", + "transformers==4.30.0", "torch>=1.1.0", "Click>=7.0.0", "tabulate",