From 85bfe4ff5cb24dda3ad5f6c73f8ada83a644c18a Mon Sep 17 00:00:00 2001
From: zqwerty <zhuq96@hotmail.com>
Date: Sat, 27 Nov 2021 03:39:43 +0000
Subject: [PATCH] update pkg requires in setup.py (torch>=1.10,
transformers>=4.0)
---
setup.py | 31 +++++++++----------------------
1 file changed, 9 insertions(+), 22 deletions(-)
diff --git a/setup.py b/setup.py
index 321a111f..799aa2b3 100755
--- a/setup.py
+++ b/setup.py
@@ -31,37 +31,24 @@ setup(
'Topic :: Scientific/Engineering :: Artificial Intelligence',
],
install_requires=[
- 'nltk>=3.4',
- 'tqdm>=4.30',
- 'checksumdir>=1.1',
- 'dataclasses',
- 'visdom',
- 'Pillow',
- 'future',
- 'torch>=1.2.0,<1.6.0',
- 'numpy>=1.15.0',
- 'scipy',
- 'scikit-learn==0.20.3',
- 'pytorch-pretrained-bert>=0.6.1',
- 'transformers>=2.3.0,<3.0.0',
- 'tensorboard>=1.14.0',
- 'tensorboardX==1.7',
- 'tokenizers>=0.8.0',
- 'overrides==4.1.2',
- 'allennlp==0.9.0',
'requests',
+ 'numpy',
+ 'nltk',
+ 'scipy',
+ 'torch>=1.10.0',
+ 'transformers>=4.0',
+ 'spacy',
+ 'allennlp',
'simplejson',
- 'spacy==2.1.9',
'unidecode',
'jieba',
'embeddings',
+ 'visdom',
'quadprog',
- 'pyyaml',
'fuzzywuzzy',
- 'python-Levenshtein',
'json_lines',
'gtts',
- 'DeepSpeech',
+ 'deepspeech',
'pydub'
],
extras_require={
--
GitLab