modify pytorch version

This commit is contained in:
tlk-dsg 2021-12-14 16:03:04 +08:00
parent 7dcc4378cb
commit cf2b207cec
1 changed files with 4 additions and 3 deletions

View File

@ -1,20 +1,21 @@
from setuptools import setup, find_packages
setup(
name='deepke', # 打包后的包文件名
version='0.2.93', #版本号
version='0.2.94', #版本号
keywords=["pip", "RE","NER","AE"], # 关键字
description='DeepKE 是基于 Pytorch 的深度学习中文关系抽取处理套件。', # 说明
long_description="client", #详细说明
license="MIT", # 许可
url='https://github.com/zjunlp/deepke',
author='ZJUNLP',
author_email='xx2020@zju.edu.cn',
author_email='zhangningyu@zju.edu.cn',
include_package_data=True,
platforms="any",
package_dir={"": "src"},
packages=find_packages("src"),
install_requires=[
'torch==1.5',
'torch>=1.5,<=1.10',
'hydra-core==1.0.6',
'tensorboard==2.4.1',
'matplotlib==3.4.1',