Skip to content
Snippets Groups Projects
Commit 85bfe4ff authored by zqwerty's avatar zqwerty
Browse files

update pkg requires in setup.py (torch>=1.10, transformers>=4.0)

parent 211d2718
Branches
No related tags found
No related merge requests found
......@@ -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={
......
0% Loading or .
You are about to add 0 people to the discussion. Proceed with caution.
Please register or to comment